Fix fine-tuning translation scripts (#9809)

This commit is contained in:
Magdalena Biesialska
2021-01-26 17:30:31 +01:00
committed by GitHub
parent c37dcff764
commit 8f6c12d306

View File

@@ -78,8 +78,8 @@ python examples/seq2seq/run_seq2seq.py \
--task translation_en_to_ro \ --task translation_en_to_ro \
--dataset_name wmt16 \ --dataset_name wmt16 \
--dataset_config_name ro-en \ --dataset_config_name ro-en \
--source_lang en-XX \ --source_lang en_XX \
--target_lang ro-RO\ --target_lang ro_RO\
--output_dir ~/tmp/tst-translation \ --output_dir ~/tmp/tst-translation \
--per_device_train_batch_size=4 \ --per_device_train_batch_size=4 \
--per_device_eval_batch_size=4 \ --per_device_eval_batch_size=4 \
@@ -96,8 +96,8 @@ python examples/seq2seq/run_seq2seq.py \
--task translation_en_to_ro \ --task translation_en_to_ro \
--dataset_name wmt16 \ --dataset_name wmt16 \
--dataset_config_name ro-en \ --dataset_config_name ro-en \
--source_lang en-XX \ --source_lang en_XX \
--target_lang ro-RO\ --target_lang ro_RO\
--train_file path_to_json_file \ --train_file path_to_json_file \
--validation_file path_to_json_file \ --validation_file path_to_json_file \
--output_dir ~/tmp/tst-translation \ --output_dir ~/tmp/tst-translation \