mems not splitted

robertmacyiii · Feb 9, 2019 · f4a07a3 · f4a07a3
1 parent 43b9af0
commit f4a07a3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/run_transfo_xl.py b/examples/run_transfo_xl.py
@@ -102,7 +102,7 @@ def evaluate(eval_iter):
         with torch.no_grad():
             mems = None
             for idx, (data, target, seq_len) in enumerate(eval_iter):
-                ret = model(data, target, *mems)
+                ret = model(data, target, mems)
                 loss, mems = ret
                 loss = loss.mean()
                 total_loss += seq_len * loss.item()