fix transformer

dmlc · Jul 29, 2020 · d9c4140 · d9c4140
1 parent e49fbe1
commit d9c4140
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/gluonnlp/models/transformer.py b/src/gluonnlp/models/transformer.py
@@ -196,7 +196,7 @@ def __init__(self,
                                        bias_initializer=bias_initializer,
                                        dtype=self._dtype)
         attention_layout = 'NTK' if self._layout == 'NT' else 'TNK'
-        self.attention_cell =\
+        self.self_attention =\
             MultiHeadAttentionCell(
                 query_units=self._units,
                 num_heads=self._num_heads,