diff --git a/configs/r50_fpn_frcnn_1x.py b/configs/r50_fpn_frcnn_1x.py index a4b9838fc4d58168a99b2cff4a2d36eb36e7e8fc..d5547ee100105348da202f5bbaff364caacf9684 100644 --- a/configs/r50_fpn_frcnn_1x.py +++ b/configs/r50_fpn_frcnn_1x.py @@ -140,7 +140,7 @@ log_config = dict( # runtime settings total_epochs = 12 device_ids = range(8) -dist_params = dict(backend='gloo') +dist_params = dict(backend='nccl') log_level = 'INFO' work_dir = './work_dirs/fpn_faster_rcnn_r50_1x' load_from = None diff --git a/configs/r50_fpn_maskrcnn_1x.py b/configs/r50_fpn_maskrcnn_1x.py index fa88de7ae318d3dc9fe0f3ff6f361f09d8b189d6..09d2fb9223102df36a2027170bc8e322b814cfd5 100644 --- a/configs/r50_fpn_maskrcnn_1x.py +++ b/configs/r50_fpn_maskrcnn_1x.py @@ -153,7 +153,7 @@ log_config = dict( # runtime settings total_epochs = 12 device_ids = range(8) -dist_params = dict(backend='gloo') +dist_params = dict(backend='nccl') log_level = 'INFO' work_dir = './work_dirs/fpn_mask_rcnn_r50_1x' load_from = None diff --git a/configs/r50_fpn_rpn_1x.py b/configs/r50_fpn_rpn_1x.py index 4c0bb41dba42e78d4d408fa170f039df79b35d7f..d2f28c2dfe6fe20bb5f64849385d712a43a40f3d 100644 --- a/configs/r50_fpn_rpn_1x.py +++ b/configs/r50_fpn_rpn_1x.py @@ -110,7 +110,7 @@ log_config = dict( # yapf:enable # runtime settings total_epochs = 12 -dist_params = dict(backend='gloo') +dist_params = dict(backend='nccl') log_level = 'INFO' work_dir = './work_dirs/fpn_rpn_r50_1x' load_from = None