Large margin softmax loss in pytroch

[will_soon (Will)]

Thanks for your effort.