mems not splitted

f4a07a39 · thomwolf · 43b9af0c · f4a07a39
Commit f4a07a39 authored Feb 09, 2019 by thomwolf
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/run_transfo_xl.py examples/run_transfo_xl.py +1 -1

No files found.
--- 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()