-
Notifications
You must be signed in to change notification settings - Fork 27
/
run.sh
executable file
·34 lines (28 loc) · 988 Bytes
/
run.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
base=nonorm_saved_components/
mkdir ${base};
save=1
class=car_normalized,motorbike_normalized,chair_normalized,bottle_normalized,airplane_normalized
# class=car1
for lr in 0.01;
do
path=${base}${class}_channels_${lr}
echo $lr $path
sbatch -c 2 --qos=tenenbaum --gres=gpu:titan-x:1 --mem=40G --time=6-12:0 -J ch${lr} main.lua \
-lr ${lr} -save_path ${path} -save_model ${save} -train_sets ${class}
done
#### lights
for lr in 0.01;
do
path=${base}${class}_lights_${lr}
echo $lr $path
sbatch -c 2 --qos=tenenbaum --gres=gpu:titan-x:1 --mem=40G --time=6-12:0 -J l${lr} lights.lua \
-lr ${lr} -save_path ${path} -save_model ${save} -train_sets ${class}
done
## shader
for lr in 0.01;
do
path=${base}${class}_shader_${lr}
echo $lr $path
sbatch -c 2 --qos=tenenbaum --gres=gpu:titan-x:1 --mem=40G --time=6-12:0 -J shad${lr} shader.lua \
-lr ${lr} -save_path ${path} -save_model ${save} -train_sets ${class}
done