Skip to content

GitLab

Unverified Commit 8eb7f26d authored by Sam Shleifer's avatar Sam Shleifer Committed by GitHub
Browse files

simplify marian distillation script (#9394)

parent d944966b
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
examples/research_projects/seq2seq-distillation/distil_marian_no_teacher.sh
View file @ 8eb7f26d
#!/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 \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment