From 98317ee6587e43f5de4a1d5918b015e098392a68 Mon Sep 17 00:00:00 2001
From: coincheung <867153576@qq.com>
Date: Thu, 4 Aug 2022 05:05:44 +0000
Subject: [PATCH] update training command

---
 dist_train.sh | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/dist_train.sh b/dist_train.sh
index e8de773..5500143 100644
--- a/dist_train.sh
+++ b/dist_train.sh
@@ -30,3 +30,17 @@ export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
 cfg_file=configs/bisenetv2_coco.py
 NGPUS=8
 torchrun --nproc_per_node=$NGPUS tools/train_amp.py --config $cfg_file
+
+
+## bisenetv1 ade20k
+export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
+cfg_file=configs/bisenetv1_ade20k.py
+NGPUS=8
+torchrun --nproc_per_node=$NGPUS tools/train_amp.py --config $cfg_file
+
+
+## bisenetv2 ade20k
+export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
+cfg_file=configs/bisenetv2_ade20k.py
+NGPUS=8
+torchrun --nproc_per_node=$NGPUS tools/train_amp.py --config $cfg_file