arcface

nnet/loss.py

@@ -91,10 +91,10 @@ def l2_norm(input,axis=1):
     return output
 
 
-class Arcface(torch.nn.Module):
+class ArcFace(torch.nn.Module):
     # implementation of additive margin softmax loss in https://arxiv.org/abs/1801.05599
     def __init__(self, embedding_size, classnum, s=64., m=0.5):
-        super(Arcface, self).__init__()
+        super(ArcFace, self).__init__()
         self.classnum = classnum
         self.kernel = Parameter(torch.Tensor(embedding_size, classnum))
         # initial kernel