Commit 12003485 authored by Kai Chen's avatar Kai Chen
Browse files

use nccl as the default comm backend

parent 6e238d52
......@@ -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
......
......@@ -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
......
......@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment