yolov5_prune_sfp icon indicating copy to clipboard operation
yolov5_prune_sfp copied to clipboard

是否需要bn层的剪枝

Open Ranking666 opened this issue 2 years ago • 1 comments

您好,非常感谢您的工作,但是我在学习您的代码时,在get_filter_codebook函数中只对卷积核的权重进行了清零,相应的bn层的权重并没有清零,这样不会导致bn层中的偏置产生一些补偿,这样会不会降低剪枝的效果?

Ranking666 avatar Apr 07 '22 09:04 Ranking666

同时,还有卷积的偏置并没有被置为0,这样是否也有影响

Ranking666 avatar Apr 07 '22 12:04 Ranking666