From 8eb7f26d5d9ce42eb88be6f0150b22a41d76a93d Mon Sep 17 00:00:00 2001
From: Sam Shleifer <sshleifer@gmail.com>
Date: Mon, 4 Jan 2021 00:51:24 -0500
Subject: [PATCH] simplify marian distillation script (#9394)

---
 .../seq2seq-distillation/distil_marian_no_teacher.sh         | 5 +++--
 1 file changed, 3 insertions(+), 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
index 4a30628149..4f0f53d796 100755
--- 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 \