Skip to content

GitLab

"mmdet3d/vscode:/vscode.git/clone" did not exist on "87339473e4538f3bcc2a9401a32347ece613c3c3"
Commit 12003485 authored by Kai Chen's avatar Kai Chen
Browse files

use nccl as the default comm backend

parent 6e238d52
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
configs/r50_fpn_frcnn_1x.py
View file @ 12003485
...@@ -140,7 +140,7 @@ log_config = dict( ...@@ -140,7 +140,7 @@ log_config = dict(
# runtime settings # runtime settings
total_epochs = 12 total_epochs = 12
device_ids = range(8) device_ids = range(8)
dist_params = dict(backend='gloo') dist_params = dict(backend='nccl')
log_level = 'INFO' log_level = 'INFO'
work_dir = './work_dirs/fpn_faster_rcnn_r50_1x' work_dir = './work_dirs/fpn_faster_rcnn_r50_1x'
load_from = None load_from = None
......
configs/r50_fpn_maskrcnn_1x.py
View file @ 12003485
...@@ -153,7 +153,7 @@ log_config = dict( ...@@ -153,7 +153,7 @@ log_config = dict(
# runtime settings # runtime settings
total_epochs = 12 total_epochs = 12
device_ids = range(8) device_ids = range(8)
dist_params = dict(backend='gloo') dist_params = dict(backend='nccl')
log_level = 'INFO' log_level = 'INFO'
work_dir = './work_dirs/fpn_mask_rcnn_r50_1x' work_dir = './work_dirs/fpn_mask_rcnn_r50_1x'
load_from = None load_from = None
......
configs/r50_fpn_rpn_1x.py
View file @ 12003485
...@@ -110,7 +110,7 @@ log_config = dict( ...@@ -110,7 +110,7 @@ log_config = dict(
# yapf:enable # yapf:enable
# runtime settings # runtime settings
total_epochs = 12 total_epochs = 12
dist_params = dict(backend='gloo') dist_params = dict(backend='nccl')
log_level = 'INFO' log_level = 'INFO'
work_dir = './work_dirs/fpn_rpn_r50_1x' work_dir = './work_dirs/fpn_rpn_r50_1x'
load_from = None 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