Update train.sh

8a8b6c14 · dengjb · e3bfda43 · 8a8b6c14
Commit 8a8b6c14 authored Nov 20, 2023 by dengjb
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 6 deletions

train.sh train.sh +2 -6

No files found.
--- a/train.sh
+++ b/train.sh
-GPU_NUM = 8
-exprot HIP_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
+export HIP_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
 export MIOPEN_BATCHNORM=1
-CONFIG='./configs/rtmdet/rtmdet_s_8xb32-300e_coco.py'
-bash ./tools/dist_train.sh \
-    ${CONFIG_FILE} \
-    ${GPU_NUM} \
+bash ./tools/dist_train.sh ./configs/rtmdet/rtmdet_s_8xb32-300e_coco.py 8