search

Eric-mingjie / network-slimming

Network Slimming (Pytorch) (ICCV 2017)
MIT License
907 stars 214 forks source link

卷积BN融合 #59

Open tangchen2 opened 3 years ago

tangchen2 commented 3 years ago

您好请教一下，ResNetV2这种BN-Relu-Conv的结构，是不是不能把所有的conv和bn都融合，比如channel selection之前的bn，他前一个conv应该是上一个shortcut两个支路相加的结果，这种情况就不能融合对嘛？

Eric-mingjie commented 3 years ago

应该是的

tangchen2 commented 3 years ago

感谢回复，试了一下bottleneck内部的conv - bn merge，提速效果收益很小