mirrors
/
mmsegmentation
mirror of https://github.com/open-mmlab/mmsegmentation.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[Fix]Use syncbn in mobilenet_v2 (#2198)

pull/2220/head
Miao Zheng 2022-10-18 17:19:37 +08:00 committed by GitHub
parent 20c7dc689c
commit e8af7a0ed0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
8 changed files with 16 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
configs/mobilenet_v2/mobilenet-v2-d8_deeplabv3_4xb2-80k_cityscapes-512x1024.py
View File

@ -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))

3
configs/mobilenet_v2/mobilenet-v2-d8_deeplabv3_4xb4-160k_ade20k-512x512.py
View File

@ -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))

3
configs/mobilenet_v2/mobilenet-v2-d8_deeplabv3plus_4xb2-80k_cityscapes-512x1024.py
View File

@ -9,6 +9,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))

3
configs/mobilenet_v2/mobilenet-v2-d8_deeplabv3plus_4xb4-160k_ade20k-512x512.py
View File

@ -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))

3
configs/mobilenet_v2/mobilenet-v2-d8_fcn_4xb2-80k_cityscapes-512x1024.py
View File

@ -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))

3
configs/mobilenet_v2/mobilenet-v2-d8_fcn_4xb4-160k_ade20k-512x512.py
View File

@ -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))

3
configs/mobilenet_v2/mobilenet-v2-d8_pspnet_4xb2-80k_cityscapes-512x1024.py
View File

@ -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))

3
configs/mobilenet_v2/mobilenet-v2-d8_pspnet_4xb4-160k_ade20k-512x512.py
View File

@ -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))