21 lines
520 B
Bash
Executable File
21 lines
520 B
Bash
Executable File
export CURRENT_DIR=${PWD}
|
|
#CNN_DIR = /home/shleifer/transformers_fork/examples/summarization/bart/cnn_dm
|
|
|
|
# Make output directory if it doesn't exist
|
|
mkdir -p $OUTPUT_DIR
|
|
|
|
# Add parent directory to python path to access lightning_base.py
|
|
export PYTHONPATH="../../":"${PYTHONPATH}"
|
|
|
|
python -m pdb -c continue finetune.py \
|
|
--data_dir=$CNN_DIR \
|
|
--teacher=bart-large-cnn \
|
|
--model_name_or_path=student \
|
|
--learning_rate=3e-5 \
|
|
--output_dir=$OUTPUT_DIR \
|
|
--do_train \
|
|
--do_predict \
|
|
--fp16 \
|
|
--val_check_interval 0.25 \
|
|
$@
|