mirrors/mmclassification
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmclassification.git synced 2025-06-03 21:53:55 +08:00
Code Issues Projects Releases Wiki Activity

Merge branch 'dev_mobilenetv2' into 'master'

Browse Source 
Update mobilenetv2

See merge request open-mmlab/mmclassification!26
This commit is contained in:
chenkai 2020-06-22 19:40:32 +08:00
commit 2a05c77f0f
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mmcls/models/backbones/mobilenet_v2.py
View File

@ -53,9 +53,7 @@ class InvertedResidual(nn.Module):
ConvModule( ConvModule(
in_channels=inplanes, in_channels=inplanes,
out_channels=hidden_dim, out_channels=hidden_dim,
kernel_size=3, kernel_size=1,
stride=1,
padding=1,
conv_cfg=conv_cfg, conv_cfg=conv_cfg,
norm_cfg=norm_cfg, norm_cfg=norm_cfg,
act_cfg=act_cfg)) act_cfg=act_cfg))
@ -74,8 +72,6 @@ class InvertedResidual(nn.Module):
in_channels=hidden_dim, in_channels=hidden_dim,
out_channels=planes, out_channels=planes,
kernel_size=1, kernel_size=1,
stride=1,
padding=0,
conv_cfg=conv_cfg, conv_cfg=conv_cfg,
norm_cfg=norm_cfg, norm_cfg=norm_cfg,
act_cfg=None) act_cfg=None)