SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

simplify marian distillation script (#9394)

Browse Source
This commit is contained in:
Sam Shleifer
2021-01-04 00:51:24 -05:00
committed by GitHub
parent d944966b19
commit 8eb7f26d5d
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
examples/research_projects/seq2seq-distillation/distil_marian_no_teacher.sh
View File

@@ -1,11 +1,12 @@
#!/usr/bin/env bash #!/usr/bin/env bash
export PYTHONPATH="../":"${PYTHONPATH}" export PYTHONPATH="../":"${PYTHONPATH}"
export WANDB_PROJECT=dmar export WANDB_PROJECT=dmar
python distillation.py \ export MAX_LEN=128
python finetune.py \
--learning_rate=3e-4 \ --learning_rate=3e-4 \
--do_train \ --do_train \
--do_predict \ --do_predict \
--fp16 --no_teacher \ --fp16 \
--val_check_interval 0.25 \ --val_check_interval 0.25 \
--data_dir $ENRO_DIR \ --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 \ --max_source_length $MAX_LEN --max_target_length $MAX_LEN --val_max_target_length $MAX_LEN --test_max_target_length $MAX_LEN \