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(
in_channels=inplanes,
out_channels=hidden_dim,
kernel_size=3,
stride=1,
padding=1,
kernel_size=1,
conv_cfg=conv_cfg,
norm_cfg=norm_cfg,
act_cfg=act_cfg))
@ -74,8 +72,6 @@ class InvertedResidual(nn.Module):
in_channels=hidden_dim,
out_channels=planes,
kernel_size=1,
stride=1,
padding=0,
conv_cfg=conv_cfg,
norm_cfg=norm_cfg,
act_cfg=None)