From 04afdb3f271c64386317f34e9276997fdf732725 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=B0=A2=E6=98=95=E8=BE=B0?= Date: Tue, 18 Oct 2022 20:13:53 +0800 Subject: [PATCH] use syncbn (#2207) --- .../mobilenet_v2/deeplabv3_m-v2-d8_512x1024_80k_cityscapes.py | 3 ++- configs/mobilenet_v2/deeplabv3_m-v2-d8_512x512_160k_ade20k.py | 3 ++- .../deeplabv3plus_m-v2-d8_512x1024_80k_cityscapes.py | 3 ++- .../mobilenet_v2/deeplabv3plus_m-v2-d8_512x512_160k_ade20k.py | 3 ++- configs/mobilenet_v2/fcn_m-v2-d8_512x1024_80k_cityscapes.py | 3 ++- configs/mobilenet_v2/fcn_m-v2-d8_512x512_160k_ade20k.py | 3 ++- configs/mobilenet_v2/pspnet_m-v2-d8_512x1024_80k_cityscapes.py | 3 ++- configs/mobilenet_v2/pspnet_m-v2-d8_512x512_160k_ade20k.py | 3 ++- 8 files changed, 16 insertions(+), 8 deletions(-) diff --git a/configs/mobilenet_v2/deeplabv3_m-v2-d8_512x1024_80k_cityscapes.py b/configs/mobilenet_v2/deeplabv3_m-v2-d8_512x1024_80k_cityscapes.py index 267483d88..949b196f5 100644 --- a/configs/mobilenet_v2/deeplabv3_m-v2-d8_512x1024_80k_cityscapes.py +++ b/configs/mobilenet_v2/deeplabv3_m-v2-d8_512x1024_80k_cityscapes.py @@ -7,6 +7,7 @@ model = dict( 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)) diff --git a/configs/mobilenet_v2/deeplabv3_m-v2-d8_512x512_160k_ade20k.py b/configs/mobilenet_v2/deeplabv3_m-v2-d8_512x512_160k_ade20k.py index e15b8cc82..6e935a622 100644 --- a/configs/mobilenet_v2/deeplabv3_m-v2-d8_512x512_160k_ade20k.py +++ b/configs/mobilenet_v2/deeplabv3_m-v2-d8_512x512_160k_ade20k.py @@ -7,6 +7,7 @@ model = dict( 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)) diff --git a/configs/mobilenet_v2/deeplabv3plus_m-v2-d8_512x1024_80k_cityscapes.py b/configs/mobilenet_v2/deeplabv3plus_m-v2-d8_512x1024_80k_cityscapes.py index d4533d79a..8529ed7c5 100644 --- a/configs/mobilenet_v2/deeplabv3plus_m-v2-d8_512x1024_80k_cityscapes.py +++ b/configs/mobilenet_v2/deeplabv3plus_m-v2-d8_512x1024_80k_cityscapes.py @@ -7,6 +7,7 @@ model = dict( 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)) diff --git a/configs/mobilenet_v2/deeplabv3plus_m-v2-d8_512x512_160k_ade20k.py b/configs/mobilenet_v2/deeplabv3plus_m-v2-d8_512x512_160k_ade20k.py index 7615a7c19..8e32e3869 100644 --- a/configs/mobilenet_v2/deeplabv3plus_m-v2-d8_512x512_160k_ade20k.py +++ b/configs/mobilenet_v2/deeplabv3plus_m-v2-d8_512x512_160k_ade20k.py @@ -7,6 +7,7 @@ model = dict( 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)) diff --git a/configs/mobilenet_v2/fcn_m-v2-d8_512x1024_80k_cityscapes.py b/configs/mobilenet_v2/fcn_m-v2-d8_512x1024_80k_cityscapes.py index a535bd0ed..263ac0254 100644 --- a/configs/mobilenet_v2/fcn_m-v2-d8_512x1024_80k_cityscapes.py +++ b/configs/mobilenet_v2/fcn_m-v2-d8_512x1024_80k_cityscapes.py @@ -7,6 +7,7 @@ model = dict( 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)) diff --git a/configs/mobilenet_v2/fcn_m-v2-d8_512x512_160k_ade20k.py b/configs/mobilenet_v2/fcn_m-v2-d8_512x512_160k_ade20k.py index c5f6ab0d6..04147cb3e 100644 --- a/configs/mobilenet_v2/fcn_m-v2-d8_512x512_160k_ade20k.py +++ b/configs/mobilenet_v2/fcn_m-v2-d8_512x512_160k_ade20k.py @@ -7,6 +7,7 @@ model = dict( 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)) diff --git a/configs/mobilenet_v2/pspnet_m-v2-d8_512x1024_80k_cityscapes.py b/configs/mobilenet_v2/pspnet_m-v2-d8_512x1024_80k_cityscapes.py index 7403bee86..9b69c516d 100644 --- a/configs/mobilenet_v2/pspnet_m-v2-d8_512x1024_80k_cityscapes.py +++ b/configs/mobilenet_v2/pspnet_m-v2-d8_512x1024_80k_cityscapes.py @@ -7,6 +7,7 @@ model = dict( 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)) diff --git a/configs/mobilenet_v2/pspnet_m-v2-d8_512x512_160k_ade20k.py b/configs/mobilenet_v2/pspnet_m-v2-d8_512x512_160k_ade20k.py index 5b72ac830..96a815351 100644 --- a/configs/mobilenet_v2/pspnet_m-v2-d8_512x512_160k_ade20k.py +++ b/configs/mobilenet_v2/pspnet_m-v2-d8_512x512_160k_ade20k.py @@ -7,6 +7,7 @@ model = dict( 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))