-
Notifications
You must be signed in to change notification settings - Fork 28
/
execute_tune_gr.sh
executable file
·41 lines (34 loc) · 1.33 KB
/
execute_tune_gr.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
34
35
36
37
38
39
40
#!/bin/bash
NUM=$1
all_files=$(ls tune_files/${NUM}/*.txt)
for file in ${all_files};
do
setting_str=$(cat ${file})
stringarray=($setting_str)
dr=${stringarray[0]}
n_sh=${stringarray[1]}
n_sp=${stringarray[2]}
nh=${stringarray[3]}
warm=${stringarray[4]}
w=${stringarray[5]}
init_scale=${stringarray[6]}
#export CONFIG=configs/patch_transformer/tune_fixed.yaml;
#export CONFIG=configs/image_transformer/bert.yaml
#export CONFIG=configs/gr_transformer/sim.yaml
#export CONFIG=configs/gr_transformer/debug_heldout.yaml
#export CONFIG=configs/gr_transformer/flip_residual.yaml
#export CONFIG=configs/gr_transformer/rotate.yaml
export CONFIG=configs/gr_transformer/stacks/recon_long_command.yaml
dir="/srv/local1/estengel/models/transformer_tune_gr_long/${dr}_${w}_${n_sh}_${n_sp}_${nh}_${warm}_${init_scale}";
export CHECKPOINT_DIR=${dir};
export ZERO_WEIGHT=${w};
export SHARED_LAYERS=${n_sh};
export SPLIT_LAYERS=${n_sp};
export NHEADS=${nh};
export WARMUP=${warm};
export INIT_SCALE=${init_scale};
echo ${CHECKPOINT_DIR} &> grid_logs/tmux_test_${NUM}
echo "executing grid_scripts in $(pwd)" >> grid_logs/tmux_test_${NUM};
bash grid_scripts/tune_transformer_gr.sh;
#echo "cuda ${CUDA_VISIBLE_DEVICE}" >> grid_logs/tmux_test_${NUM}
done;