search

Eric-mingjie / network-slimming

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

剪枝后的网络参数量如何统计 #43

Open 18651626101 opened 4 years ago

18651626101 commented 4 years ago

您好！： 我想请问一下，对于剪枝过后的网络，您是用何种方法统计其参数量的呢？因为按照我个人的理解，代码其实是把一些不重要的权重置零？ 谢谢！

Eric-mingjie commented 4 years ago

mask implementation是把不重要的权重置零，但我们给的基本的实现是真正实现channel pruning,尽管每个res block的第一个convolution用了channel selection layer采取零置零，但这是为了保证主干道channel 数固定的必须操作。