renamed examples

e77721e4 · thomwolf · 009b5813 · e77721e4 · e77721e4
Commit e77721e4 authored Feb 07, 2019 by thomwolf
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/train_openai_gpt.py examples/train_openai_gpt.py +0 -0

examples/train_transfo_xl.py examples/train_transfo_xl.py +1 -1

No files found.
--- a/examples/openai_gpt_train.py
+++ b/examples/openai_gpt_train.py
--- a/examples/transfo_xl_train.py
+++ b/examples/transfo_xl_train.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