Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
train.sh 202 Bytes
Newer Older
dengjb's avatar
update code  
dengjb committed
1
2
3
4
5
6
7 
GPU_NUM = 8
exprot 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} \