search

Eric-mingjie / network-slimming

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

mask-impl fine-tune #24

Open colorjam opened 4 years ago

colorjam commented 4 years ago

你好，mask版本的prune_mask并没有真正对权重进行剪枝，在finetune时，虽然用的是mask后的权重，但是对所有parameters进行更新，这样bn那些为0的权重不是也被训练了吗？

Eric-mingjie commented 4 years ago

https://github.com/Eric-mingjie/network-slimming/blob/master/mask-impl/main_mask.py#L131 并没有对全部parameter进行更新

250zhanghu commented 2 years ago

https://github.com/Eric-mingjie/network-slimming/blob/master/mask-impl/main_mask.py#L131 并没有对全部parameter进行更新 你好大佬，这个没有对剪枝掉的bn通道更新，但是它对应通道的卷积层参数还是更新了吧。有没有方法可以使被剪掉通道的bn和卷积参数都不更新呢。十分感谢！