From e848b54730de9f5421337e3074f33b2e16d52a3a Mon Sep 17 00:00:00 2001 From: thomwolf Date: Tue, 16 Jul 2019 21:22:19 +0200 Subject: [PATCH] fix #792 --- examples/single_model_scripts/run_transfo_xl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/single_model_scripts/run_transfo_xl.py b/examples/single_model_scripts/run_transfo_xl.py index fda0d8dc28..95efbb8855 100644 --- a/examples/single_model_scripts/run_transfo_xl.py +++ b/examples/single_model_scripts/run_transfo_xl.py @@ -114,7 +114,7 @@ def main(): mems = None for idx, (data, target, seq_len) in enumerate(eval_iter): ret = model(data, target, mems) - loss, mems = ret + loss, _, mems = ret loss = loss.mean() total_loss += seq_len * loss.item() total_len += seq_len