fix argument order (#3637)

This commit is contained in:
Patrick von Platen
2020-04-05 12:33:41 +02:00
committed by GitHub
parent b809d2f073
commit 1789c7daf1

View File

@@ -488,7 +488,7 @@ class TFTransfoXLMainLayer(tf.keras.layers.Layer):
else:
return None
def _update_mems(self, hids, mems, qlen, mlen):
def _update_mems(self, hids, mems, mlen, qlen):
# does not deal with None
if mems is None:
return None