[pl_examples] default warmup steps=0 (#5316)

2020-06-26 15:03:41 -04:00
parent bf0d12c220
commit 5543b30aa6
6 changed files with 14 additions and 13 deletions
--- a/examples/seq2seq/train_distilbart_cnn.sh
+++ b/examples/seq2seq/train_distilbart_cnn.sh
@@ -16,9 +16,9 @@ python finetune.py \
    --freeze_encoder --freeze_embeds --data_dir $CNN_DIR \
    --max_target_length 142 --val_max_target_length=142 \
    --train_batch_size=$BS --eval_batch_size=$BS --gradient_accumulation_steps=$GAS \
-    --data_dir $CNN_DIR \
    --model_name_or_path sshleifer/student_cnn_12_6 \
    --tokenizer_name facebook/bart-large \
+    --warmup_steps 500 \
    --output_dir distilbart-cnn-12-6 \
    $@