xiaolai-sqlai / mobilenetv3

mobilenetv3 with pytorch,provide pre-train model
MIT License
1.6k stars 340 forks source link

SE模块位置与论文中不符? #51

Closed Yangworld4 closed 1 month ago

Yangworld4 commented 1 month ago

源码是不是有问题?我在看论文中与别人的实现代码中发现SE模块是在DeptWise之后使用,其中的输入通道数是expand_size也就是扩大的通道数,随后进行PointWise,而此代码是在PointWise结束后进行SE机制。

如图为其它实现代码 微信图片_20240510010107