From a1cb6e98667e9c35abf7f19e611f3c61f7c98f0f Mon Sep 17 00:00:00 2001 From: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> Date: Wed, 23 Dec 2020 11:05:21 -0500 Subject: [PATCH] Adapt to new name of `label_smoothing_factor` training arg (#9282) --- examples/seq2seq/test_finetune_trainer.py | 2 +- examples/seq2seq/train_distil_marian_enro.sh | 2 +- examples/seq2seq/train_distil_marian_enro_tpu.sh | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/seq2seq/test_finetune_trainer.py b/examples/seq2seq/test_finetune_trainer.py index e8f7d431b6..9490462971 100644 --- a/examples/seq2seq/test_finetune_trainer.py +++ b/examples/seq2seq/test_finetune_trainer.py @@ -153,7 +153,7 @@ class TestFinetuneTrainer(TestCasePlus): --save_steps {str(eval_steps)} --eval_steps {str(eval_steps)} --sortish_sampler - --label_smoothing 0.1 + --label_smoothing_factor 0.1 --adafactor --task translation --tgt_lang ro_RO diff --git a/examples/seq2seq/train_distil_marian_enro.sh b/examples/seq2seq/train_distil_marian_enro.sh index 82f7725003..fc1b90595c 100644 --- a/examples/seq2seq/train_distil_marian_enro.sh +++ b/examples/seq2seq/train_distil_marian_enro.sh @@ -34,5 +34,5 @@ python finetune_trainer.py \ --do_train --do_eval --do_predict \ --evaluation_strategy steps \ --predict_with_generate --logging_first_step \ - --task translation --label_smoothing 0.1 \ + --task translation --label_smoothing_factor 0.1 \ "$@" diff --git a/examples/seq2seq/train_distil_marian_enro_tpu.sh b/examples/seq2seq/train_distil_marian_enro_tpu.sh index 82f6ce1406..2fce7684ab 100644 --- a/examples/seq2seq/train_distil_marian_enro_tpu.sh +++ b/examples/seq2seq/train_distil_marian_enro_tpu.sh @@ -35,5 +35,5 @@ python xla_spawn.py --num_cores $TPU_NUM_CORES \ --do_train --do_eval \ --evaluation_strategy steps \ --prediction_loss_only \ - --task translation --label_smoothing 0.1 \ + --task translation --label_smoothing_factor 0.1 \ "$@"