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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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