dsrg/train.sh at main · joosun7l/dsrg · GitHub

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
#!/bin/bash
name=1
arch=deeplab_large_fov
resume='./datalist/PascalVOC/vgg16_20M_custom.pth'
dataset="PascalVOC"
data='/home/joosunki/shared/VOC/VOCdevkit/VOC2012'
gt_root='./datalist/PascalVOC/localization_cues.pickle'
train_list='./datalist/PascalVOC/input_list.txt'
batch=16
gamma=0.1
max_iter=8000
stepsize=2000
snapshot=500
wd=0.0005
lr=0.0005
thre_fg=0.85
thre_bg=0.99
gpu=0

python3.8 main.py \
    --arch ${arch} \
    --name ${name} \
    --data ${data} \
    --gt-root ${gt_root} \
    --dataset ${dataset} \
    --train-list ${train_list} \
    --max-iter ${max_iter} \
    --snapshot ${snapshot} \
    --lr-decay ${stepsize} \
    --batch-size ${batch} \
    --lr ${lr} \
    --wd ${wd} \
    --resume ${resume} \
    --thre-bg ${thre_bg} \
    --thre-fg ${thre_fg} \
    --gpu ${gpu}