transformers/examples/seq2seq/run_distiller.sh
2020-08-17 09:06:35 -04:00

11 lines
224 B
Bash
Executable File

# Add parent directory to python path to access lightning_base.py
export PYTHONPATH="../":"${PYTHONPATH}"
python distillation.py \
--learning_rate=3e-4 \
--do_train \
--do_predict \
--fp16 \
--val_check_interval 0.1 \
"$@"