fix _update_memory fn call in transformer-xl (#2971)

This commit is contained in:
Sandro Cavallari
2020-02-24 17:50:24 -05:00
committed by GitHub
parent 6a50d501ec
commit ee60840ee6
+1 -1
View File
@@ -645,7 +645,7 @@ class TransfoXLModel(TransfoXLPreTrainedModel):
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