-
Notifications
You must be signed in to change notification settings - Fork 0
/
run_sc_2.sh
executable file
·46 lines (43 loc) · 1.53 KB
/
run_sc_2.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
41
42
43
44
45
46
#!/bin/bash
dataset_name=$1
data_path=$2
likelihood_n=$3
epochs=$4
h_dim=$5
depth=$6
norm=$7
# for prod_space in "e2"
# do
# echo "----- Training: sc -----"
# python -m mt.examples.run --dataset=$dataset_name \
# --data=$data_path \
# --model=$prod_space \
# --fixed_curvature=True \
# --h_dim=$h_dim \
# --depth=$depth \
# --norm=$norm \
# --architecture="ff" \
# --likelihood_n=$likelihood_n \
# --epochs=$epochs \
# --warmup=50 \
# --lookahead=25 \
# --device="cuda:1"
# done
for prod_space in "e4" "e6"
do
echo "----- Training: sc -----"
python -m mt.examples.run --dataset=$dataset_name \
--data=$data_path \
--model=$prod_space \
--fixed_curvature=True \
--h_dim=$h_dim \
--depth=$depth \
--norm=$norm \
--architecture="ff" \
--likelihood_n=$likelihood_n \
--epochs=$epochs \
--warmup=100 \
--lookahead=50 \
--device="cuda:1" \
--batch_size=300
done