-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathexperiments_mnist_ses.sh
45 lines (39 loc) · 1.07 KB
/
experiments_mnist_ses.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
# !/bin/bash
# MIT License. Copyright (c) 2020 Ivan Sosnovik, Michał Szmaja
MNIST_SCALE_DIR="${MNIST_SCALE_DIR:-./datasets}"
function train_scale_mnist() {
# 1 model_name
# 2 extra_scaling
# for seed in {0..5}
for seed in {0..0}
do
python train_scale_mnist.py \
--batch_size 128 \
--epochs 60 \
--optim adam \
--lr 0.01 \
--lr_steps 20 40 \
--model $1 \
--basis "G"\
--save_model_path "./saved_models/mnist/$1_extra_scaling_$2.pt" \
--cuda \
--extra_scaling $2 \
--tag "sesn_experiments" \
done
}
#model_list=(
# "mnist_ses_scalar_28" # MNIST (28x28)
# "mnist_ses_vector_28" # MNIST (28x28)
# "mnist_ses_scalar_28p" # MNIST (28x28) +
# "mnist_ses_vector_28p" # MNIST (28x28) +
#)
model_list=(
"mnist_ses_vector_96_rot_8_interrot_4"
)
for model_name in "${model_list[@]}"
do
for extra_scaling in 1.0
do
train_scale_mnist "$model_name" "$extra_scaling"
done
done