From 43b9af0cac55c2bcde11ddea8658ada6f1bb1872 Mon Sep 17 00:00:00 2001 From: thomwolf Date: Sat, 9 Feb 2019 16:12:19 +0100 Subject: [PATCH] mems initialized to None in run_transfo --- 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 1218a1f547..3a0a70e68e 100644 --- a/examples/run_transfo_xl.py +++ b/examples/run_transfo_xl.py @@ -100,7 +100,7 @@ def main(): total_len, total_loss = 0, 0. start_time = time.time() with torch.no_grad(): - mems = tuple() + mems = None for idx, (data, target, seq_len) in enumerate(eval_iter): ret = model(data, target, *mems) loss, mems = ret