Update gasoftmaxloss.py

ydwen · May 14, 2022 · 032c31b · 032c31b
1 parent 3119c97
commit 032c31b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/model/head/gasoftmaxloss.py b/model/head/gasoftmaxloss.py
@@ -7,6 +7,7 @@
 
 class GAsoftmax(nn.Module):
     """ reference: <Deep Hyperspherical Learning> in NIPS 2017"
+        Section 3 in the paper: https://arxiv.org/abs/1711.03189
     """
     def __init__(self, feat_dim, num_class, s=30., m=1.5):
         super(GAsoftmax, self).__init__()