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

[s2s test_finetune_trainer] failing multigpu test (#8400)

Browse Source
This commit is contained in:
Stas Bekman
2020-11-08 13:45:40 -08:00
committed by GitHub
parent f62755a600
commit 66582492d3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/seq2seq/test_finetune_trainer.py
View File

@@ -177,7 +177,7 @@ class TestFinetuneTrainer(TestCasePlus):
--num_train_epochs {str(num_train_epochs)} --num_train_epochs {str(num_train_epochs)}
--per_device_train_batch_size 4 --per_device_train_batch_size 4
--per_device_eval_batch_size 4 --per_device_eval_batch_size 4
--learning_rate 3e-4 --learning_rate 3e-3
--warmup_steps 8 --warmup_steps 8
--evaluate_during_training --evaluate_during_training
--predict_with_generate --predict_with_generate