增加如下代码即可

  • 前边的输入参数添加​​DEBUG​​：
parser.add_argument('--DEBUG', type=bool, default=True, help='debug mode')
  • 代码中间加入分布式环境变量：
if args.DEBUG:
        args.launcher = 'pytorch'
        os.environ['LOCAL_RANK'] = str(0)
        os.environ['RANK'] = str(0)
        os.environ['WORLD_SIZE'] = str(get_dist_info()[1])
        os.environ['MASTER_ADDR'] = '127.0.0.1'
        os.environ['MASTER_PORT'] = str(25004)

具体位置如下：

MMDet——用单卡train.py debug分布式代码_python