Skip to content

GitLab

Commit e77721e4 authored by thomwolf's avatar thomwolf
Browse files

renamed examples

parent 009b5813
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/transfo_xl_train.py → examples/train_transfo_xl.py
View file @ e77721e4
...@@ -437,7 +437,7 @@ def evaluate(eval_iter): ...@@ -437,7 +437,7 @@ def evaluate(eval_iter):
if args.max_eval_steps > 0 and i >= args.max_eval_steps: if args.max_eval_steps > 0 and i >= args.max_eval_steps:
break break
ret = model(data, target, *mems) ret = model(data, target, *mems)
loss, mems = ret[0], ret[1:] loss, mems = ret
loss = loss.mean() loss = loss.mean()
total_loss += seq_len * loss.float().item() total_loss += seq_len * loss.float().item()
total_len += seq_len total_len += seq_len
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment