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