[s2s test_finetune_trainer] failing multigpu test (huggingface#8400)

stas00 · Nov 10, 2020 · 2e69f7f · 2e69f7f
1 parent f0365c8
commit 2e69f7f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/seq2seq/test_finetune_trainer.py b/examples/seq2seq/test_finetune_trainer.py
@@ -177,7 +177,7 @@ def run_trainer(self, eval_steps: int, max_len: str, model_name: str, num_train_
             --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