renamed examples

2019-02-07 23:15:15 +01:00
parent 009b581316
commit e77721e4fe
2 changed files with 1 additions and 1 deletions
--- a/examples/train_openai_gpt.py
+++ b/examples/train_openai_gpt.py
--- a/examples/train_transfo_xl.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