From f4a07a392c769dbc66dada3d3b6ac9e1761f89da Mon Sep 17 00:00:00 2001 From: thomwolf Date: Sat, 9 Feb 2019 16:14:31 +0100 Subject: [PATCH] mems not splitted --- examples/run_transfo_xl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/run_transfo_xl.py b/examples/run_transfo_xl.py index 3a0a70e68e..b8000a2080 100644 --- 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()