use syncbn (open-mmlab#2207)

Author: xiexinch

configs/mobilenet_v2/deeplabv3_m-v2-d8_512x1024_80k_cityscapes.py

@@ -7,6 +7,7 @@
         widen_factor=1.,
         strides=(1, 2, 2, 1, 1, 1, 1),
         dilations=(1, 1, 1, 2, 2, 4, 4),
-        out_indices=(1, 2, 4, 6)),
+        out_indices=(1, 2, 4, 6),
+        norm_cfg=dict(type='SyncBN', requires_grad=True)),
     decode_head=dict(in_channels=320),
     auxiliary_head=dict(in_channels=96))

configs/mobilenet_v2/deeplabv3_m-v2-d8_512x512_160k_ade20k.py

@@ -7,6 +7,7 @@
         widen_factor=1.,
         strides=(1, 2, 2, 1, 1, 1, 1),
         dilations=(1, 1, 1, 2, 2, 4, 4),
-        out_indices=(1, 2, 4, 6)),
+        out_indices=(1, 2, 4, 6),
+        norm_cfg=dict(type='SyncBN', requires_grad=True)),
     decode_head=dict(in_channels=320),
     auxiliary_head=dict(in_channels=96))

configs/mobilenet_v2/deeplabv3plus_m-v2-d8_512x1024_80k_cityscapes.py

@@ -7,6 +7,7 @@
         widen_factor=1.,
         strides=(1, 2, 2, 1, 1, 1, 1),
         dilations=(1, 1, 1, 2, 2, 4, 4),
-        out_indices=(1, 2, 4, 6)),
+        out_indices=(1, 2, 4, 6),
+        norm_cfg=dict(type='SyncBN', requires_grad=True)),
     decode_head=dict(in_channels=320, c1_in_channels=24),
     auxiliary_head=dict(in_channels=96))

configs/mobilenet_v2/deeplabv3plus_m-v2-d8_512x512_160k_ade20k.py

@@ -7,6 +7,7 @@
         widen_factor=1.,
         strides=(1, 2, 2, 1, 1, 1, 1),
         dilations=(1, 1, 1, 2, 2, 4, 4),
-        out_indices=(1, 2, 4, 6)),
+        out_indices=(1, 2, 4, 6),
+        norm_cfg=dict(type='SyncBN', requires_grad=True)),
     decode_head=dict(in_channels=320, c1_in_channels=24),
     auxiliary_head=dict(in_channels=96))

configs/mobilenet_v2/fcn_m-v2-d8_512x1024_80k_cityscapes.py

@@ -7,6 +7,7 @@
         widen_factor=1.,
         strides=(1, 2, 2, 1, 1, 1, 1),
         dilations=(1, 1, 1, 2, 2, 4, 4),
-        out_indices=(1, 2, 4, 6)),
+        out_indices=(1, 2, 4, 6),
+        norm_cfg=dict(type='SyncBN', requires_grad=True)),
     decode_head=dict(in_channels=320),
     auxiliary_head=dict(in_channels=96))

configs/mobilenet_v2/fcn_m-v2-d8_512x512_160k_ade20k.py

@@ -7,6 +7,7 @@
         widen_factor=1.,
         strides=(1, 2, 2, 1, 1, 1, 1),
         dilations=(1, 1, 1, 2, 2, 4, 4),
-        out_indices=(1, 2, 4, 6)),
+        out_indices=(1, 2, 4, 6),
+        norm_cfg=dict(type='SyncBN', requires_grad=True)),
     decode_head=dict(in_channels=320),
     auxiliary_head=dict(in_channels=96))

configs/mobilenet_v2/pspnet_m-v2-d8_512x1024_80k_cityscapes.py

@@ -7,6 +7,7 @@
         widen_factor=1.,
         strides=(1, 2, 2, 1, 1, 1, 1),
         dilations=(1, 1, 1, 2, 2, 4, 4),
-        out_indices=(1, 2, 4, 6)),
+        out_indices=(1, 2, 4, 6),
+        norm_cfg=dict(type='SyncBN', requires_grad=True)),
     decode_head=dict(in_channels=320),
     auxiliary_head=dict(in_channels=96))

configs/mobilenet_v2/pspnet_m-v2-d8_512x512_160k_ade20k.py

@@ -7,6 +7,7 @@
         widen_factor=1.,
         strides=(1, 2, 2, 1, 1, 1, 1),
         dilations=(1, 1, 1, 2, 2, 4, 4),
-        out_indices=(1, 2, 4, 6)),
+        out_indices=(1, 2, 4, 6),
+        norm_cfg=dict(type='SyncBN', requires_grad=True)),
     decode_head=dict(in_channels=320),
     auxiliary_head=dict(in_channels=96))