-
Notifications
You must be signed in to change notification settings - Fork 23
Expand file tree
/
Copy pathrun_lrgb_mpnn_slurm.sh
More file actions
189 lines (134 loc) · 11.2 KB
/
run_lrgb_mpnn_slurm.sh
File metadata and controls
189 lines (134 loc) · 11.2 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
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
#!/usr/bin/env bash
# Run this script from the project root dir.
function run_repeats {
dataset=$1
cfg_suffix=$2
# The cmd line cfg overrides that will be passed to the main.py,
# e.g. 'name_tag test01 gnn.layer_type gcnconv'
cfg_overrides=$3
cfg_file="${cfg_dir}/${dataset}-${cfg_suffix}.yaml"
if [[ ! -f "$cfg_file" ]]; then
echo "WARNING: Config does not exist: $cfg_file"
echo "SKIPPING!"
return 1
fi
main="python main.py --cfg ${cfg_file}"
out_dir="results/${dataset}" # <-- Set the output dir.
common_params="out_dir ${out_dir} ${cfg_overrides}"
echo "Run program: ${main}"
echo " output dir: ${out_dir}"
# Run each repeat as a separate job
for SEED in {0..3}; do
# script="sbatch ${slurm_directive} -J ${cfg_suffix}-${dataset} run/wrapper.sb ${main} --repeat 1 seed ${SEED} ${common_params}"
script="sbatch ${slurm_directive} -J ${cfg_suffix}-${dataset} run/wrapper-narval.sb ${main} --repeat 1 seed ${SEED} ${common_params}"
echo $script
eval $script
done
}
echo "Do you wish to sbatch jobs? Assuming this is the project root dir: `pwd`"
select yn in "Yes" "No"; do
case $yn in
Yes ) break;;
No ) exit;;
esac
done
################################################################################
##### GCN
################################################################################
## 1.1 VOCSuperpixels
cfg_dir="configs/GCN"
slurm_directive="--time=0-11:59:00 --mem=16G --gres=gpu:1 --cpus-per-task=4"
# 1.1.1 GCN VOCSuperpixels slic 10
run_repeats vocsuperpixels GCN "wandb.project lrgb-voc name_tag GCN-VOC-COOgraph-slic10 dataset.name edge_wt_only_coord dataset.slic_compactness 10"
run_repeats vocsuperpixels GCN "wandb.project lrgb-voc name_tag GCN-VOC-COOFEATgraph-slic10 dataset.name edge_wt_coord_feat dataset.slic_compactness 10"
run_repeats vocsuperpixels GCN "wandb.project lrgb-voc name_tag GCN-VOC-RBgraph-slic10 dataset.name edge_wt_region_boundary dataset.slic_compactness 10"
# 1.1.2 GCN VOCSuperpixels slic 30
run_repeats vocsuperpixels GCN "wandb.project lrgb-voc name_tag GCN-VOC-COOgraph-slic30 dataset.name edge_wt_only_coord dataset.slic_compactness 30"
run_repeats vocsuperpixels GCN "wandb.project lrgb-voc name_tag GCN-VOC-COOFEATgraph-slic30 dataset.name edge_wt_coord_feat dataset.slic_compactness 30"
run_repeats vocsuperpixels GCN "wandb.project lrgb-voc name_tag GCN-VOC-RBgraph-slic30 dataset.name edge_wt_region_boundary dataset.slic_compactness 30"
## 1.2 COCOSuperpixels
cfg_dir="configs/GCN"
slurm_directive="--time=2-00:00:00 --mem=64G --gres=gpu:1 --cpus-per-task=4"
# 1.2.1 GCN COCOSuperpixels slic 10
run_repeats cocosuperpixels GCN "wandb.project lrgb-coco name_tag GCN-COCO-COOgraph-slic10 dataset.name edge_wt_only_coord dataset.slic_compactness 10"
run_repeats cocosuperpixels GCN "wandb.project lrgb-coco name_tag GCN-COCO-COOFEATgraph-slic10 dataset.name edge_wt_coord_feat dataset.slic_compactness 10"
run_repeats cocosuperpixels GCN "wandb.project lrgb-coco name_tag GCN-COCO-RBgraph-slic10 dataset.name edge_wt_region_boundary dataset.slic_compactness 10"
# 1.2.2 GCN COCOSuperpixels slic 30
run_repeats cocosuperpixels GCN "wandb.project lrgb-coco name_tag GCN-COCO-COOgraph-slic30 dataset.name edge_wt_only_coord dataset.slic_compactness 30"
run_repeats cocosuperpixels GCN "wandb.project lrgb-coco name_tag GCN-COCO-COOFEATgraph-slic30 dataset.name edge_wt_coord_feat dataset.slic_compactness 30"
run_repeats cocosuperpixels GCN "wandb.project lrgb-coco name_tag GCN-COCO-RBgraph-slic30 dataset.name edge_wt_region_boundary dataset.slic_compactness 30"
################################################################################
##### GINE
################################################################################
## 1.1 VOCSuperpixels
cfg_dir="configs/GINE"
slurm_directive="--time=0-15:59:00 --mem=16G --gres=gpu:1 --cpus-per-task=4"
# 1.1.1 GINE VOCSuperpixels slic 10
run_repeats vocsuperpixels GINE "wandb.project lrgb-voc name_tag GINE-VOC-COOgraph-slic10 dataset.name edge_wt_only_coord dataset.slic_compactness 10"
run_repeats vocsuperpixels GINE "wandb.project lrgb-voc name_tag GINE-VOC-COOFEATgraph-slic10 dataset.name edge_wt_coord_feat dataset.slic_compactness 10"
run_repeats vocsuperpixels GINE "wandb.project lrgb-voc name_tag GINE-VOC-RBgraph-slic10 dataset.name edge_wt_region_boundary dataset.slic_compactness 10"
# 1.1.2 GINE VOCSuperpixels slic 30
run_repeats vocsuperpixels GINE "wandb.project lrgb-voc name_tag GINE-VOC-COOgraph-slic30 dataset.name edge_wt_only_coord dataset.slic_compactness 30"
run_repeats vocsuperpixels GINE "wandb.project lrgb-voc name_tag GINE-VOC-COOFEATgraph-slic30 dataset.name edge_wt_coord_feat dataset.slic_compactness 30"
run_repeats vocsuperpixels GINE "wandb.project lrgb-voc name_tag GINE-VOC-RBgraph-slic30 dataset.name edge_wt_region_boundary dataset.slic_compactness 30"
## 1.2 COCOSuperpixels
cfg_dir="configs/GINE"
slurm_directive="--time=2-10:00:00 --mem=64G --gres=gpu:1 --cpus-per-task=4"
# 1.2.1 GINE COCOSuperpixels slic 10
run_repeats cocosuperpixels GINE "wandb.project lrgb-coco name_tag GINE-COCO-COOgraph-slic10 dataset.name edge_wt_only_coord dataset.slic_compactness 10"
run_repeats cocosuperpixels GINE "wandb.project lrgb-coco name_tag GINE-COCO-COOFEATgraph-slic10 dataset.name edge_wt_coord_feat dataset.slic_compactness 10"
run_repeats cocosuperpixels GINE "wandb.project lrgb-coco name_tag GINE-COCO-RBgraph-slic10 dataset.name edge_wt_region_boundary dataset.slic_compactness 10"
# 1.2.2 GINE COCOSuperpixels slic 30
run_repeats cocosuperpixels GINE "wandb.project lrgb-coco name_tag GINE-COCO-COOgraph-slic30 dataset.name edge_wt_only_coord dataset.slic_compactness 30"
run_repeats cocosuperpixels GINE "wandb.project lrgb-coco name_tag GINE-COCO-COOFEATgraph-slic30 dataset.name edge_wt_coord_feat dataset.slic_compactness 30"
run_repeats cocosuperpixels GINE "wandb.project lrgb-coco name_tag GINE-COCO-RBgraph-slic30 dataset.name edge_wt_region_boundary dataset.slic_compactness 30"
################################################################################
##### GatedGCN
################################################################################
## 1.1 VOCSuperpixels
cfg_dir="configs/GatedGCN"
slurm_directive="--time=0-11:59:00 --mem=16G --gres=gpu:1 --cpus-per-task=4"
# 1.1.1 GatedGCN VOCSuperpixels slic 10
run_repeats vocsuperpixels GatedGCN "wandb.project lrgb-voc name_tag GatedGCN-VOC-COOgraph-slic10 dataset.name edge_wt_only_coord dataset.slic_compactness 10"
run_repeats vocsuperpixels GatedGCN "wandb.project lrgb-voc name_tag GatedGCN-VOC-COOFEATgraph-slic10 dataset.name edge_wt_coord_feat dataset.slic_compactness 10"
run_repeats vocsuperpixels GatedGCN "wandb.project lrgb-voc name_tag GatedGCN-VOC-RBgraph-slic10 dataset.name edge_wt_region_boundary dataset.slic_compactness 10"
# 1.1.2 GatedGCN VOCSuperpixels slic 30
run_repeats vocsuperpixels GatedGCN "wandb.project lrgb-voc name_tag GatedGCN-VOC-COOgraph-slic30 dataset.name edge_wt_only_coord dataset.slic_compactness 30"
run_repeats vocsuperpixels GatedGCN "wandb.project lrgb-voc name_tag GatedGCN-VOC-COOFEATgraph-slic30 dataset.name edge_wt_coord_feat dataset.slic_compactness 30"
run_repeats vocsuperpixels GatedGCN "wandb.project lrgb-voc name_tag GatedGCN-VOC-RBgraph-slic30 dataset.name edge_wt_region_boundary dataset.slic_compactness 30"
## 1.2 COCOSuperpixels
cfg_dir="configs/GatedGCN"
slurm_directive="--time=2-00:00:00 --mem=64G --gres=gpu:1 --cpus-per-task=4"
# 1.2.1 GatedGCN COCOSuperpixels slic 10
run_repeats cocosuperpixels GatedGCN "wandb.project lrgb-coco name_tag GatedGCN-COCO-COOgraph-slic10 dataset.name edge_wt_only_coord dataset.slic_compactness 10"
run_repeats cocosuperpixels GatedGCN "wandb.project lrgb-coco name_tag GatedGCN-COCO-COOFEATgraph-slic10 dataset.name edge_wt_coord_feat dataset.slic_compactness 10"
run_repeats cocosuperpixels GatedGCN "wandb.project lrgb-coco name_tag GatedGCN-COCO-RBgraph-slic10 dataset.name edge_wt_region_boundary dataset.slic_compactness 10"
# 1.2.2 GatedGCN COCOSuperpixels slic 30
run_repeats cocosuperpixels GatedGCN "wandb.project lrgb-coco name_tag GatedGCN-COCO-COOgraph-slic30 dataset.name edge_wt_only_coord dataset.slic_compactness 30"
run_repeats cocosuperpixels GatedGCN "wandb.project lrgb-coco name_tag GatedGCN-COCO-COOFEATgraph-slic30 dataset.name edge_wt_coord_feat dataset.slic_compactness 30"
run_repeats cocosuperpixels GatedGCN "wandb.project lrgb-coco name_tag GatedGCN-COCO-RBgraph-slic30 dataset.name edge_wt_region_boundary dataset.slic_compactness 30"
################################################################################
##### GatedGCN-LapPE
################################################################################
## 1.1 VOCSuperpixels
cfg_dir="configs/GatedGCN"
slurm_directive="--time=0-11:59:00 --mem=16G --gres=gpu:1 --cpus-per-task=4"
# 1.1.1 GatedGCN-LapPE VOCSuperpixels slic 10
run_repeats vocsuperpixels GatedGCN+LapPE "wandb.project lrgb-voc name_tag GatedGCN-LapPE-VOC-COOgraph-slic10 dataset.name edge_wt_only_coord dataset.slic_compactness 10"
run_repeats vocsuperpixels GatedGCN+LapPE "wandb.project lrgb-voc name_tag GatedGCN-LapPE-VOC-COOFEATgraph-slic10 dataset.name edge_wt_coord_feat dataset.slic_compactness 10"
run_repeats vocsuperpixels GatedGCN+LapPE "wandb.project lrgb-voc name_tag GatedGCN-LapPE-VOC-RBgraph-slic10 dataset.name edge_wt_region_boundary dataset.slic_compactness 10"
# 1.1.2 GatedGCN-LapPE VOCSuperpixels slic 30
run_repeats vocsuperpixels GatedGCN+LapPE "wandb.project lrgb-voc name_tag GatedGCN-LapPE-VOC-COOgraph-slic30 dataset.name edge_wt_only_coord dataset.slic_compactness 30"
run_repeats vocsuperpixels GatedGCN+LapPE "wandb.project lrgb-voc name_tag GatedGCN-LapPE-VOC-COOFEATgraph-slic30 dataset.name edge_wt_coord_feat dataset.slic_compactness 30"
run_repeats vocsuperpixels GatedGCN+LapPE "wandb.project lrgb-voc name_tag GatedGCN-LapPE-VOC-RBgraph-slic30 dataset.name edge_wt_region_boundary dataset.slic_compactness 30"
## 1.2 COCOSuperpixels
cfg_dir="configs/GatedGCN"
slurm_directive="--time=2-00:00:00 --mem=64G --gres=gpu:1 --cpus-per-task=4"
# 1.2.1 GatedGCN-LapPE COCOSuperpixels slic 10
run_repeats cocosuperpixels GatedGCN+LapPE "wandb.project lrgb-coco name_tag GatedGCN-LapPE-COCO-COOgraph-slic10 dataset.name edge_wt_only_coord dataset.slic_compactness 10"
run_repeats cocosuperpixels GatedGCN+LapPE "wandb.project lrgb-coco name_tag GatedGCN-LapPE-COCO-COOFEATgraph-slic10 dataset.name edge_wt_coord_feat dataset.slic_compactness 10"
run_repeats cocosuperpixels GatedGCN+LapPE "wandb.project lrgb-coco name_tag GatedGCN-LapPE-COCO-RBgraph-slic10 dataset.name edge_wt_region_boundary dataset.slic_compactness 10"
# 1.2.2 GatedGCN-LapPE COCOSuperpixels slic 30
run_repeats cocosuperpixels GatedGCN+LapPE "wandb.project lrgb-coco name_tag GatedGCN-LapPE-COCO-COOgraph-slic30 dataset.name edge_wt_only_coord dataset.slic_compactness 30"
run_repeats cocosuperpixels GatedGCN+LapPE "wandb.project lrgb-coco name_tag GatedGCN-LapPE-COCO-COOFEATgraph-slic30 dataset.name edge_wt_coord_feat dataset.slic_compactness 30"
run_repeats cocosuperpixels GatedGCN+LapPE "wandb.project lrgb-coco name_tag GatedGCN-LapPE-COCO-RBgraph-slic30 dataset.name edge_wt_region_boundary dataset.slic_compactness 30"