Giters

xhwNobody / yolov5_prune_sfp

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

是否需要bn层的剪枝

Ranking666 opened this issue · comments

commented

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

commented

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