Skip to content

Commit 98317ee

Browse files
committed
update training command
1 parent 184dc5a commit 98317ee

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

dist_train.sh

+14
Original file line numberDiff line numberDiff line change
@@ -30,3 +30,17 @@ export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
3030
cfg_file=configs/bisenetv2_coco.py
3131
NGPUS=8
3232
torchrun --nproc_per_node=$NGPUS tools/train_amp.py --config $cfg_file
33+
34+
35+
## bisenetv1 ade20k
36+
export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
37+
cfg_file=configs/bisenetv1_ade20k.py
38+
NGPUS=8
39+
torchrun --nproc_per_node=$NGPUS tools/train_amp.py --config $cfg_file
40+
41+
42+
## bisenetv2 ade20k
43+
export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
44+
cfg_file=configs/bisenetv2_ade20k.py
45+
NGPUS=8
46+
torchrun --nproc_per_node=$NGPUS tools/train_amp.py --config $cfg_file

0 commit comments

Comments
 (0)