Open colorjam opened 4 years ago
你好,mask版本的prune_mask并没有真正对权重进行剪枝,在finetune时,虽然用的是mask后的权重,但是对所有parameters进行更新,这样bn那些为0的权重不是也被训练了吗?
https://github.com/Eric-mingjie/network-slimming/blob/master/mask-impl/main_mask.py#L131 并没有对全部parameter进行更新
https://github.com/Eric-mingjie/network-slimming/blob/master/mask-impl/main_mask.py#L131 并没有对全部parameter进行更新 你好大佬,这个没有对剪枝掉的bn通道更新,但是它对应通道的卷积层参数还是更新了吧。有没有方法可以使被剪掉通道的bn和卷积参数都不更新呢。十分感谢!
你好,mask版本的prune_mask并没有真正对权重进行剪枝,在finetune时,虽然用的是mask后的权重,但是对所有parameters进行更新,这样bn那些为0的权重不是也被训练了吗?