-
Notifications
You must be signed in to change notification settings - Fork 1
/
run.sh
executable file
·60 lines (54 loc) · 1.12 KB
/
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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
python main.py \
--arch=vgg19 \
--mode=lowrank \
--batch-size=128 \
--epochs=300 \
--full-rank-warmup=True \
--fr-warmup-epoch=30 \
--seed=42 \
--lr=0.1 \
--resume=False \
--evaluate=False \
--ckpt_path=./checkpoint/resnet18_best.pth \
--momentum=0.9 > log 2>&1
# python main.py \
# --arch=vgg19 \
# --mode=vanilla \
# --batch-size=128 \
# --epochs=300 \
# --full-rank-warmup=False \
# --fr-warmup-epoch=30 \
# --seed=42 \
# --lr=0.1 \
# --resume=False \
# --evaluate=False \
# --ckpt_path=./checkpoint/resnet18_best.pth \
# --momentum=0.9 > log 2>&1
# python main.py \
# --arch=resnet18 \
# --mode=lowrank \
# --batch-size=128 \
# --epochs=300 \
# --full-rank-warmup=True \
# --fr-warmup-epoch=80 \
# --seed=42 \
# --evaluate=True \
# --resume=True \
# --ckpt_path=./checkpoint/resnet18_best.pth \
# --lr=0.1 \
# --momentum=0.9
for SEED in 1 2 3
do
python main.py \
--arch=resnet18 \
--mode=lowrank \
--batch-size=128 \
--epochs=300 \
--full-rank-warmup=True \
--fr-warmup-epoch=80 \
--seed=${SEED} \
--lr=0.1 \
--resume=False \
--evaluate=False \
--momentum=0.9 > pufferfish_resnet18_cifar10_seed${SEED}_log 2>&1
done