[s2s test_finetune_trainer] failing multigpu test (#8400)

2020-11-08 13:45:40 -08:00
parent f62755a600
commit 66582492d3
1 changed files with 1 additions and 1 deletions
--- a/examples/seq2seq/test_finetune_trainer.py
+++ b/examples/seq2seq/test_finetune_trainer.py
@@ -177,7 +177,7 @@ class TestFinetuneTrainer(TestCasePlus):
            --num_train_epochs {str(num_train_epochs)}
            --per_device_train_batch_size 4
            --per_device_eval_batch_size 4
-            --learning_rate 3e-4
+            --learning_rate 3e-3
            --warmup_steps 8
            --evaluate_during_training
            --predict_with_generate