Skip to content
Snippets Groups Projects
Commit 12003485 authored by Kai Chen's avatar Kai Chen
Browse files

use nccl as the default comm backend

parent 6e238d52
No related branches found
No related tags found
No related merge requests found
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment