mems not splitted

2019-02-09 16:14:31 +01:00
parent 43b9af0cac
commit f4a07a392c
1 changed files with 1 additions and 1 deletions
--- a/examples/run_transfo_xl.py
+++ b/examples/run_transfo_xl.py
@@ -102,7 +102,7 @@ def main():
        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()