/
run_moco_track2.sh
41 lines (38 loc) · 1.15 KB
/
run_moco_track2.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
#!/usr/bin/env bash
export root_path='./data/' ;
python3 ./moco/main_moco_pretraining_phase2.py \
-a densenet121 \
--workers 32 \
--lr 0.03 \
--batch-size 128 \
--epochs 200 \
--input-size 224 \
--dist-url 'tcp://localhost:10001' --multiprocessing-distributed --world-size 1 --rank 0 \
--gpus 0,1,2,3 \
--mlp --moco-t 0.2 --moco-k 4096 --aug-plus --cos \
--track track2 \
"${root_path}public_ood_0412_nodomainlabel/train/" ;
python3 ./moco/main_moco_pretraining_phase2.py \
-a resnet34 \
--workers 32 \
--lr 0.03 \
--batch-size 128 \
--epochs 200 \
--input-size 224 \
--dist-url 'tcp://localhost:10001' --multiprocessing-distributed --world-size 1 --rank 0 \
--gpus 0,1,2,3 \
--mlp --moco-t 0.2 --moco-k 4096 --aug-plus --cos \
--track track2 \
"${root_path}public_ood_0412_nodomainlabel/train/" ;
python3 ./moco/main_moco_pretraining_phase2.py \
-a wide_resnet50_2 \
--workers 32 \
--lr 0.03 \
--batch-size 128 \
--epochs 200 \
--input-size 224 \
--dist-url 'tcp://localhost:10001' --multiprocessing-distributed --world-size 1 --rank 0 \
--gpus 0,1,2,3 \
--mlp --moco-t 0.2 --moco-k 4096 --aug-plus --cos \
--track track2 \
"${root_path}public_ood_0412_nodomainlabel/train/" ;