forked from henrikmarklund/arm
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathall_commands.sh
More file actions
121 lines (106 loc) · 5.19 KB
/
all_commands.sh
File metadata and controls
121 lines (106 loc) · 5.19 KB
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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
# MNIST
SEEDS="0 1 2"
SHARED_ARGS="\
--dataset mnist \
--num_epochs 200 \
--eval_on val test \
--n_samples_per_group 300 \
--seeds ${SEEDS} \
--meta_batch_size 6 \
--epochs_per_eval 10 \
--optimizer adam \
--learning_rate 1e-4 \
--weight_decay 0 \
--log_wandb 0"
N_CONTEXT_CHANNELS=12 # For CML
python run.py --exp_name erm ${SHARED_ARGS}
python run.py --exp_name uw --uniform_over_groups 1 ${SHARED_ARGS}
python run.py --exp_name drnn --algorithm DRNN --uniform_over_groups 1 $SHARED_ARGS
python run.py --exp_name mmd --algorithm MMD --sampler group --uniform_over_groups 1 $SHARED_ARGS
python run.py --exp_name dann --algorithm DANN $SHARED_ARGS
python run.py --exp_name arm-cml --algorithm ARM-CML --sampler group --uniform_over_groups 1 --n_context_channels $N_CONTEXT_CHANNELS $SHARED_ARGS
python run.py --exp_name arm-ll --algorithm ARM-LL --sampler group --uniform_over_groups 1 $SHARED_ARGS
python run.py --exp_name arm-bn --algorithm ARM-BN --sampler group --uniform_over_groups 1 $SHARED_ARGS
python run.py --exp_name cml-ablation --algorithm ARM-CML --n_context_channels $N_CONTEXT_CHANNELS $SHARED_ARGS
python run.py --exp_name ll-ablation --algorithm ARM-LL $SHARED_ARGS
python run.py --exp_name bn-ablation --algorithm ARM-BN $SHARED_ARGS
# FEMNIST
SEEDS="0 1 2"
SHARED_ARGS="\
--dataset femnist \
--num_epochs 200 \
--eval_on val test \
--seeds ${SEEDS} \
--meta_batch_size 2 \
--epochs_per_eval 1 \
--optimizer sgd \
--learning_rate 1e-4 \
--weight_decay 1e-4 \
--log_wandb 0"
N_CONTEXT_CHANNELS=1 # For CML
python run.py --exp_name erm ${SHARED_ARGS}
python run.py --exp_name uw --uniform_over_groups 1 ${SHARED_ARGS}
python run.py --exp_name drnn --algorithm DRNN --uniform_over_groups 1 $SHARED_ARGS
python run.py --exp_name mmd --algorithm MMD --sampler group --uniform_over_groups 1 $SHARED_ARGS
#python run.py --exp_name dann --algorithm DANN $SHARED_ARGS # run separately with optimizer=adam
python run.py --exp_name arm-cml --algorithm ARM-CML --sampler group --uniform_over_groups 1 --n_context_channels $N_CONTEXT_CHANNELS $SHARED_ARGS
python run.py --exp_name arm-ll --algorithm ARM-LL --sampler group --uniform_over_groups 1 $SHARED_ARGS
python run.py --exp_name arm-bn --algorithm ARM-BN --sampler group --uniform_over_groups 1 $SHARED_ARGS
python run.py --exp_name cml-ablation --algorithm ARM-CML --n_context_channels $N_CONTEXT_CHANNELS $SHARED_ARGS
python run.py --exp_name ll-ablation --algorithm ARM-LL $SHARED_ARGS
python run.py --exp_name bn-ablation --algorithm ARM-BN $SHARED_ARGS
# CIFAR-C
SEEDS="0 1 2"
SHARED_ARGS="\
--dataset cifar-c \
--num_epochs 100 \
--n_samples_per_group 2000 \
--test_n_samples_per_group 3000 \
--eval_on val test \
--seeds ${SEEDS} \
--meta_batch_size 3 \
--support_size 100 \
--epochs_per_eval 1 \
--optimizer sgd \
--learning_rate 1e-2 \
--weight_decay 1e-4 \
--log_wandb 0"
N_CONTEXT_CHANNELS=3 # For CML
python run.py --exp_name erm ${SHARED_ARGS}
python run.py --exp_name drnn --algorithm DRNN --uniform_over_groups 1 $SHARED_ARGS
python run.py --exp_name mmd --algorithm MMD --sampler group --uniform_over_groups 1 $SHARED_ARGS
python run.py --exp_name dann --algorithm DANN $SHARED_ARGS
python run.py --exp_name arm-cml --algorithm ARM-CML --sampler group --uniform_over_groups 1 --n_context_channels $N_CONTEXT_CHANNELS --adapt_bn 1 $SHARED_ARGS
python run.py --exp_name arm-ll --algorithm ARM-LL --sampler group --uniform_over_groups 1 $SHARED_ARGS
python run.py --exp_name arm-bn --algorithm ARM-BN --sampler group --uniform_over_groups 1 $SHARED_ARGS
python run.py --exp_name cml-ablation --algorithm ARM-CML --n_context_channels --adapt_bn 1 $N_CONTEXT_CHANNELS $SHARED_ARGS
python run.py --exp_name ll-ablation --algorithm ARM-LL $SHARED_ARGS
python run.py --exp_name bn-ablation --algorithm ARM-BN $SHARED_ARGS
# Tiny ImageNet-C
SEEDS="0 1 2"
SHARED_ARGS="\
--dataset tinyimg \
--num_epochs 50 \
--n_samples_per_group 2000 \
--test_n_samples_per_group 3000 \
--eval_on val test \
--seeds ${SEEDS} \
--meta_batch_size 3 \
--support_size 100 \
--model resnet50 \
--epochs_per_eval 1 \
--optimizer sgd \
--learning_rate 1e-2 \
--weight_decay 1e-4 \
--log_wandb 0"
N_CONTEXT_CHANNELS=3 # For CML
python run.py --exp_name erm ${SHARED_ARGS}
python run.py --exp_name drnn --algorithm DRNN --uniform_over_groups 1 $SHARED_ARGS
python run.py --exp_name mmd --algorithm MMD --sampler group --uniform_over_groups 1 $SHARED_ARGS
python run.py --exp_name dann --algorithm DANN $SHARED_ARGS
python run.py --exp_name arm-cml --algorithm ARM-CML --sampler group --uniform_over_groups 1 --n_context_channels $N_CONTEXT_CHANNELS --adapt_bn 1 $SHARED_ARGS
python run.py --exp_name arm-ll --algorithm ARM-LL --sampler group --uniform_over_groups 1 $SHARED_ARGS
python run.py --exp_name arm-bn --algorithm ARM-BN --sampler group --uniform_over_groups 1 $SHARED_ARGS
python run.py --exp_name cml-ablation --algorithm ARM-CML --n_context_channels $N_CONTEXT_CHANNELS --adapt_bn 1 $SHARED_ARGS
python run.py --exp_name ll-ablation --algorithm ARM-LL $SHARED_ARGS
python run.py --exp_name bn-ablation --algorithm ARM-BN $SHARED_ARGS