Update attention_keras.py

Eadon999 · Jul 27, 2019 · b7415c9 · b7415c9
1 parent 661e4e4
commit b7415c9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/attention_keras.py b/attention_keras.py
@@ -426,7 +426,7 @@ class SinCosPositionEmbedding(Layer):
     """
     def __init__(self, v_dim,
                  merge_mode='add', **kwargs):
-        super(CosSinPositionEmbedding, self).__init__(**kwargs)
+        super(SinCosPositionEmbedding, self).__init__(**kwargs)
         self.v_dim = v_dim
         self.merge_mode = merge_mode
     def call(self, inputs):