增加如下代码即可

• 前边的输入参数添加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)

具体位置如下：