From 12003485b30382f3c0d522a040ba1e76cf3e0a93 Mon Sep 17 00:00:00 2001
From: Kai Chen <chenkaidev@gmail.com>
Date: Mon, 8 Oct 2018 19:29:51 +0800
Subject: [PATCH] use nccl as the default comm backend

---
 configs/r50_fpn_frcnn_1x.py    | 2 +-
 configs/r50_fpn_maskrcnn_1x.py | 2 +-
 configs/r50_fpn_rpn_1x.py      | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/configs/r50_fpn_frcnn_1x.py b/configs/r50_fpn_frcnn_1x.py
index a4b9838..d5547ee 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 fa88de7..09d2fb9 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 4c0bb41..d2f28c2 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
-- 
GitLab