这里采用多GPU训练,这里以2个GPUs为例,可以在 tools/dist_train.sh中进行GPU设置如export CUDA_VISIBLE_DEVICES=0,1,然后执行:
bash tools/dist_train.sh configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py 2

这里采用多GPU训练,这里以2个GPUs为例,可以在 tools/dist_train.sh中进行GPU设置如export CUDA_VISIBLE_DEVICES=0,1,然后执行:
bash tools/dist_train.sh configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py 2