-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathfewrl_concat_dev.sh
33 lines (31 loc) · 1.66 KB
/
fewrl_concat_dev.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
#!/bin/bash
seeds=(12321 943 111 300 1300 545 12 10001 77 1993)
for i in ${!seeds[@]};
do
seed=${seeds[$i]}
gsutil -m rsync -r gs://emnlp-2022-rebuttal/fewrel-concat/fewrel/concat_run_${seed}/ ~/sep-1/fewrel/concat_run_${seed}/
for (( e=0; e<=4; e++ ))
do
for (( s=1; s<=105; s++ ))
do
step=$((s * 100))
printf "step ${step} on epoch ${e} for seed ${seed}\r\n"
CUDA_VISIBLE_DEVICES=0 python3.7 src/re_gold_qa_train.py \
--mode concat_fewrl_dev \
--model_path ~/sep-1/fewrel/concat_run_${seed}/ \
--checkpoint _${e}_step_${step}_model \
--learning_rate 0.0005 \
--batch_size 128 \
--gpu True \
--train ./fewrl_data/train_data_${seed}.csv \
--dev ./fewrl_data/val_data_${seed}.csv \
--test ./fewrl_data/test_data_${seed}.csv \
--gpu_device 0 \
--prediction_file ~/sep-1/fewrel/concat_run_${seed}/relation.concat.run.${seed}.epoch.${e}.dev.predictions.step.${step}.csv \
--predict_type relation \
--seed ${seed}
done
done
gsutil -m rsync -r ~/sep-1/fewrel/concat_run_${seed}/ gs://emnlp-2022-rebuttal/fewrel-concat/fewrel/concat_run_${seed}/
rm -r -f ~/sep-1/fewrel/concat_run_${seed}/model*
done