对 resnet18进行剪枝

剪枝的主要流程

使用pytorch进行网络修剪

Other

已经有大神把resnet18的剪枝方法,发出来了,提到BatchNorm层通道数修改,当所有卷积层剪枝结束,依据邻近上一个卷积层输出通道数,通过BatchNorm层继承方式,它需修改成同样的通道数。但是通用的方法是把BatchNorm和上一层conv进行融合,其中会直接删除一些结构,需要一点时间,好好看看才能实现。

参考:
大佬 jacobgil
对 resnet18进行剪枝
小网络剪枝的demo
pytorch基于卷积层通道剪枝的方法
基于Pytorch的卷积神经网络剪枝

发布了1636 篇原创文章 · 获赞 341 · 访问量 221万+

猜你喜欢

转载自blog.csdn.net/tony2278/article/details/103904662
今日推荐