-
Notifications
You must be signed in to change notification settings - Fork 12
Expand file tree
/
Copy pathcoco_stuff_164K.sh
More file actions
27 lines (25 loc) · 1.02 KB
/
coco_stuff_164K.sh
File metadata and controls
27 lines (25 loc) · 1.02 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
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 \
python -m torch.distributed.launch --master_addr 127.5.0.9 --master_port 2601 --nproc_per_node=8 \
train_baseline.py \
--model deeplabv3 \
--backbone resnet50_original \
--dataset coco_stuff_164k \
--data [your dataset path]/coco_stuff_164k/ \
--batch-size 16 \
--workers 8 \
--crop-size 512 512 \
--lr 0.02 \
--max-iterations 40000 \
--save-dir [your directory path to store checkpoint files] \
--log-dir [your directory path to store log files] \
--val-per-iters 40000 \
--pretrained-base [your pretrained-backbone path]/resnet50_mixskd.pth
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 \
python -m torch.distributed.launch --master_addr 127.5.0.9 --master_port 2601 --nproc_per_node=8 \
eval.py \
--model deeplabv3 \
--backbone resnet50_original \
--dataset coco_stuff_164k \
--data [your dataset path]/coco_stuff_164k/ \
--save-dir [your directory path to store checkpoint files] \
--pretrained [your pretrained model path]