REPRODUCTION.md
October 11, 2021 ยท View on GitHub
Train with multiple GPUs
./tools/dist_train.sh ${CONFIG_FILE} ${GPU_NUM}
e.g.,
CUDA_VISIBLE_DEVICES=0,1,2,3 ./tools/dist_train.sh ./configs/sgnet/sgnet_r50_caffe_fpn_gn_1x.py 4
October 11, 2021 ยท View on GitHub
./tools/dist_train.sh ${CONFIG_FILE} ${GPU_NUM}
e.g.,
CUDA_VISIBLE_DEVICES=0,1,2,3 ./tools/dist_train.sh ./configs/sgnet/sgnet_r50_caffe_fpn_gn_1x.py 4