diff --git a/examples/seq2seq/test_finetune_trainer.py b/examples/seq2seq/test_finetune_trainer.py index 6da0e240c41959..399c1b6c047e8c 100644 --- 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