From 66582492d35edd2cd929dad8d668c982fa617211 Mon Sep 17 00:00:00 2001 From: Stas Bekman Date: Sun, 8 Nov 2020 13:45:40 -0800 Subject: [PATCH] [s2s test_finetune_trainer] failing multigpu test (#8400) --- examples/seq2seq/test_finetune_trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/seq2seq/test_finetune_trainer.py b/examples/seq2seq/test_finetune_trainer.py index 6da0e240c4..399c1b6c04 100644 --- 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