Merge pull request #41 from YuxinZou/dist

add some missing configurations
Media-Smart · Oct 16, 2020 · d656f8a · d656f8a
2 parents 7e982b5 + 7ff894d
commit d656f8a
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 2 deletions.
diff --git a/configs/coco_multilabel_unet.py b/configs/coco_multilabel_unet.py
@@ -210,7 +210,8 @@
     metrics=[
         dict(type='MultiLabelIoU', num_classes=nclasses),
         dict(type='MultiLabelMIoU', num_classes=nclasses),
-    ]
+    ],
+    dist_params=dict(backend='nccl'),
 )
 
 ## 2.1 configuration for test

diff --git a/configs/coco_unet.py b/configs/coco_unet.py
@@ -211,7 +211,8 @@
     metrics=[
         dict(type='IoU', num_classes=nclasses),
         dict(type='MIoU', num_classes=nclasses, average='equal'),
-    ]
+    ],
+    dist_params=dict(backend='nccl'),
 )
 
 ## 2.1 configuration for test

diff --git a/configs/voc_fpn.py b/configs/voc_fpn.py
@@ -215,6 +215,7 @@
         dict(type='IoU', num_classes=nclasses),
         dict(type='MIoU', num_classes=nclasses, average='equal'),
     ],
+    dist_params=dict(backend='nccl'),
 )
 
 ## 2.1 configuration for test

diff --git a/configs/voc_pspnet.py b/configs/voc_pspnet.py
@@ -81,6 +81,7 @@
         dict(type='IoU', num_classes=nclasses),
         dict(type='MIoU', num_classes=nclasses, average='equal'),
     ],
+    dist_params=dict(backend='nccl'),
 )
 
 ## 2.1 configuration for test

diff --git a/configs/voc_unet.py b/configs/voc_unet.py
@@ -210,6 +210,7 @@
         dict(type='IoU', num_classes=nclasses),
         dict(type='MIoU', num_classes=nclasses, average='equal'),
     ],
+    dist_params=dict(backend='nccl'),
 )
 
 ## 2.1 configuration for test