add some missing configurations

Media-Smart · Oct 16, 2020 · 7ff894d · 7ff894d
1 parent c67a74b
commit 7ff894d
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