From e77721e4fe8f2665132bf11ab26c3ab352e2f2a2 Mon Sep 17 00:00:00 2001 From: thomwolf Date: Thu, 7 Feb 2019 23:15:15 +0100 Subject: [PATCH] renamed examples --- examples/{openai_gpt_train.py => train_openai_gpt.py} | 0 examples/{transfo_xl_train.py => train_transfo_xl.py} | 2 +- 2 files changed, 1 insertion(+), 1 deletion(-) rename examples/{openai_gpt_train.py => train_openai_gpt.py} (100%) rename examples/{transfo_xl_train.py => train_transfo_xl.py} (99%) diff --git a/examples/openai_gpt_train.py b/examples/train_openai_gpt.py similarity index 100% rename from examples/openai_gpt_train.py rename to examples/train_openai_gpt.py diff --git a/examples/transfo_xl_train.py b/examples/train_transfo_xl.py similarity index 99% rename from examples/transfo_xl_train.py rename to examples/train_transfo_xl.py index 09d30aed28..6ea0920489 100644 --- a/examples/transfo_xl_train.py +++ b/examples/train_transfo_xl.py @@ -437,7 +437,7 @@ def evaluate(eval_iter): if args.max_eval_steps > 0 and i >= args.max_eval_steps: break ret = model(data, target, *mems) - loss, mems = ret[0], ret[1:] + loss, mems = ret loss = loss.mean() total_loss += seq_len * loss.float().item() total_len += seq_len