Skip to content

GitLab

Unverified Commit 66582492 authored by Stas Bekman's avatar Stas Bekman Committed by GitHub
Browse files

[s2s test_finetune_trainer] failing multigpu test (#8400)

parent f62755a6
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/seq2seq/test_finetune_trainer.py
View file @ 66582492
...@@ -177,7 +177,7 @@ class TestFinetuneTrainer(TestCasePlus): ...@@ -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
......
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