Update amsoftmax.py

hao-qiang · Sep 29, 2018 · 84676d1 · 84676d1
1 parent a304d2d
commit 84676d1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/amsoftmax.py b/amsoftmax.py
@@ -7,7 +7,7 @@
 import tensorflow as tf
 import numpy as np
 
-class Cosine(Layer):
+class AMSoftmax(Layer):
     def __init__(self, output_dim, **kwargs):
         self.output_dim = output_dim
         super(AMSoftmax, self).__init__(**kwargs)
@@ -56,4 +56,4 @@ def amsoftmax_loss(y_true, y_pred):
 
     cross_ent = tf.nn.softmax_cross_entropy_with_logits(labels=y_true, logits=added_embeddingFeature)
     loss = tf.reduce_mean(cross_ent)
-    return loss
+    return loss