simplify marian distillation script (huggingface#9394)

guyrosin · Jan 15, 2021 · a13cd4c · a13cd4c
1 parent 38b2a73
commit a13cd4c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/examples/research_projects/seq2seq-distillation/distil_marian_no_teacher.sh b/examples/research_projects/seq2seq-distillation/distil_marian_no_teacher.sh
@@ -1,11 +1,12 @@
 #!/usr/bin/env bash
 export PYTHONPATH="../":"${PYTHONPATH}"
 export WANDB_PROJECT=dmar
-python distillation.py \
+export MAX_LEN=128
+python finetune.py \
   --learning_rate=3e-4 \
   --do_train \
   --do_predict \
-  --fp16 --no_teacher \
+  --fp16 \
   --val_check_interval 0.25 \
   --data_dir $ENRO_DIR \
   --max_source_length $MAX_LEN --max_target_length $MAX_LEN --val_max_target_length $MAX_LEN --test_max_target_length $MAX_LEN \