-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtrain_msls_vicreg_final_sbatch.sh
executable file
·35 lines (29 loc) · 3.34 KB
/
train_msls_vicreg_final_sbatch.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
#!/bin/bash
# Random vs partial
sbatch --export=ALL,SSL=vicreg,LR=1e-5,BATCH=64,PROJ=2048,LAY=2,NEG=0,FC=1024,NEGQ=0 ./script/train_msls_ssl_long_vicreg_neg_random_pair.sbatch
sbatch --export=ALL,SSL=bt,LR=1e-5,BATCH=64,PROJ=2048,LAY=2,NEG=0,FC=1024,NEGQ=0 ./script/train_msls_ssl_long_vicreg_neg_random_pair.sbatch
sbatch --export=ALL,SSL=vicreg,LR=1e-5,BATCH=64,PROJ=2048,LAY=2,NEG=1000,FC=1024,NEGQ=1 ./script/train_msls_ssl_long_vicreg_neg_partial_pair_aug.sbatch
sbatch --export=ALL,SSL=bt,LR=1e-5,BATCH=64,PROJ=2048,LAY=2,NEG=1000,FC=1024,NEGQ=1 ./script/train_msls_ssl_long_vicreg_neg_partial_pair_aug.sbatch
# Different number of layers
sbatch --export=ALL,SSL=vicreg,LR=1e-5,BATCH=64,PROJ=2048,LAY=0,NEG=1000,FC=1024,NEGQ=1 ./script/train_msls_ssl_long_vicreg_neg_partial_pair_aug.sbatch
sbatch --export=ALL,SSL=bt,LR=1e-5,BATCH=64,PROJ=2048,LAY=0,NEG=1000,FC=1024,NEGQ=1 ./script/train_msls_ssl_long_vicreg_neg_partial_pair_aug.sbatch
sbatch --export=ALL,SSL=vicreg,LR=1e-5,BATCH=64,PROJ=2048,LAY=1,NEG=1000,FC=1024,NEGQ=1 ./script/train_msls_ssl_long_vicreg_neg_partial_pair_aug.sbatch
sbatch --export=ALL,SSL=bt,LR=1e-5,BATCH=64,PROJ=2048,LAY=1,NEG=1000,FC=1024,NEGQ=1 ./script/train_msls_ssl_long_vicreg_neg_partial_pair_aug.sbatch
sbatch --export=ALL,SSL=vicreg,LR=1e-5,BATCH=64,PROJ=2048,LAY=3,NEG=1000,FC=1024,NEGQ=1 ./script/train_msls_ssl_long_vicreg_neg_partial_pair_aug.sbatch
sbatch --export=ALL,SSL=bt,LR=1e-5,BATCH=64,PROJ=2048,LAY=3,NEG=1000,FC=1024,NEGQ=1 ./script/train_msls_ssl_long_vicreg_neg_partial_pair_aug.sbatch
# Different number of proj dim
sbatch --export=ALL,SSL=vicreg,LR=1e-5,BATCH=64,PROJ=4096,LAY=2,NEG=1000,FC=1024,NEGQ=1 ./script/train_msls_ssl_long_vicreg_neg_partial_pair_aug.sbatch
sbatch --export=ALL,SSL=bt,LR=1e-5,BATCH=64,PROJ=4096,LAY=2,NEG=1000,FC=1024,NEGQ=1 ./script/train_msls_ssl_long_vicreg_neg_partial_pair_aug.sbatch
sbatch --export=ALL,SSL=vicreg,LR=1e-5,BATCH=64,PROJ=1024,LAY=2,NEG=1000,FC=1024,NEGQ=1 ./script/train_msls_ssl_long_vicreg_neg_partial_pair_aug.sbatch
sbatch --export=ALL,SSL=bt,LR=1e-5,BATCH=64,PROJ=1024,LAY=2,NEG=1000,FC=1024,NEGQ=1 ./script/train_msls_ssl_long_vicreg_neg_partial_pair_aug.sbatch
# Different number of fc dim
sbatch --export=ALL,SSL=vicreg,LR=1e-5,BATCH=64,PROJ=2048,LAY=2,NEG=1000,FC=512,NEGQ=1 ./script/train_msls_ssl_long_vicreg_neg_partial_pair_aug.sbatch
sbatch --export=ALL,SSL=bt,LR=1e-5,BATCH=64,PROJ=2048,LAY=2,NEG=1000,FC=512,NEGQ=1 ./script/train_msls_ssl_long_vicreg_neg_partial_pair_aug.sbatch
sbatch --export=ALL,SSL=vicreg,LR=1e-5,BATCH=64,PROJ=2048,LAY=2,NEG=1000,FC=2048,NEGQ=1 ./script/train_msls_ssl_long_vicreg_neg_partial_pair_aug.sbatch
sbatch --export=ALL,SSL=bt,LR=1e-5,BATCH=64,PROJ=2048,LAY=2,NEG=1000,FC=2048,NEGQ=1 ./script/train_msls_ssl_long_vicreg_neg_partial_pair_aug.sbatch
# Final
sbatch --export=ALL,SSL=vicreg,LR=1e-5,BATCH=64,PROJ=4096,LAY=2,NEG=1000,FC=1024,NEGQ=1 ./script/train_msls_ssl_long_vicreg_neg_partial_pair_aug.sbatch
sbatch --export=ALL,SSL=bt,LR=1e-5,BATCH=64,PROJ=2048,LAY=2,NEG=1000,FC=1024,NEGQ=1 ./script/train_msls_ssl_long_vicreg_neg_partial_pair_aug.sbatch
# Deit
sbatch --export=ALL,SSL=bt,LR=1e-5,BATCH=32,PROJ=2048,LAY=2,NEG=1000,FC=256,NEGQ=1 ./script/train_msls_ssl_long_vicreg_neg_partial_pair_deit.sbatch
sbatch --export=ALL,SSL=vicreg,LR=1e-5,BATCH=32,PROJ=4096,LAY=2,NEG=1000,FC=256,NEGQ=1 ./script/train_msls_ssl_long_vicreg_neg_partial_pair_deit.sbatch