Skip to content

Commit 858d351

Browse files
committed
para update
1 parent 3a69136 commit 858d351

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

run_voc.sh

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,8 @@ NAME=EXP_VOC
44
EXP_DIR=$HOME/CoSA_EXP
55
RANDPORT=6531
66
echo Using port $RANDPORT
7-
torchrun --master_port $RANDPORT --nproc_per_node=1 main.py $NAME --work_dir $EXP_DIR \
7+
torchrun --master_port $RANDPORT --nproc_per_node=2 main.py $NAME --work_dir $EXP_DIR \
88
--dataset VOC12 \
99
--voc12_root $HOME/data/VOCdevkit/VOC2012 \
10+
--max_iters 32000 \
11+
--aux_layer -4 \

0 commit comments

Comments
 (0)