SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

renamed examples

Browse Source
This commit is contained in:
thomwolf
2019-02-07 23:15:15 +01:00
parent 009b581316
commit e77721e4fe
2 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

0
examples/openai_gpt_train.py → examples/train_openai_gpt.py
View File

2
examples/transfo_xl_train.py → examples/train_transfo_xl.py
View File

@@ -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