xiaolai-sqlai / mobilenetv3

mobilenetv3 with pytorch,provide pre-train model

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

SE's position differs from paper.

TKONIY opened this issue · comments

image
image

SE should be placed after Depthwise and before Pointwise. However, you set it after Pointwise. Is there any reason or just a mistake?

@TKONIY This is one of several problems, see an earlier issue #8

I changed the place of SE and I put before out = self.bn3(self.conv3(out)) but it makes an error

@rebeen It is not enough. Check the SE class and network class, we can see that SE takes a parameter of input channels, which should match the output channels of dw. It should be modified because now it matches the output channels of pw.

Thank you very much I will try to tackle this issue