/
run_training.sh
executable file
·18 lines (10 loc) · 1.01 KB
/
run_training.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
dataset=$1
checkpoint=$2
python scripts/neg/finetune.py -bs 128 -d $dataset -t -ckpt $checkpoint -c scripts/neg/configs/t5_vae_32_max_none_none.yaml &&
python scripts/neg/finetune.py -bs 128 -d $dataset -t -ckpt $checkpoint -c scripts/neg/configs/t5_vae_32_max_none_0.15.yaml &&
python scripts/neg/finetune.py -bs 128 -d $dataset -t -ckpt $checkpoint -c scripts/neg/configs/t5_vae_32_max_none_0.4.yaml &&
python scripts/neg/finetune.py -bs 128 -d $dataset -t -ckpt $checkpoint -c scripts/neg/configs/t5_vae_32_max_0.5_none.yaml &&
python scripts/neg/finetune.py -bs 128 -d $dataset -t -ckpt $checkpoint -c scripts/neg/configs/t5_vae_32_max_0.5_0.15.yaml &&
python scripts/neg/finetune.py -bs 128 -d $dataset -t -ckpt $checkpoint -c scripts/neg/configs/t5_vae_32_max_0.5_0.4.yaml &&
python scripts/neg/finetune.py -bs 128 -d $dataset -t -ckpt $checkpoint -c scripts/neg/configs/t5_vae_32_max_3_0.15.yaml &&
python scripts/neg/finetune.py -bs 128 -d $dataset -t -ckpt $checkpoint -c scripts/neg/configs/t5_vae_32_max_6_0.15.yaml