0
点赞
收藏
分享

微信扫一扫

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


增加如下代码即可

  • 前边的输入参数添加​​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分布式代码_环境变量


举报

相关推荐

0 条评论