search

ZJU-lishuang / yolov5_prune

yolov5 prune，Support V2, V3, V4 and V6 versions of yolov5
Apache License 2.0
556 stars 136 forks source link

请问剪枝策略一中的，prune_model_but_keepsize这块是什么原理呀？看了半天代码没看懂 #102

Open dddmmmyyy1998 opened 2 years ago

dddmmmyyy1998 commented 2 years ago

https://github.com/ZJU-lishuang/yolov5_prune/blob/126054962197a51c79140384c591b9190d146019/utils/prune_utils.py#L471

求大佬指点一二

ZJU-lishuang commented 2 years ago

Learning Efficient Convolutional Networks Through Network Slimming (ICCV 2017)

ZJU-lishuang commented 2 years ago

函数的作用应该是更新剪枝后的激活参数

dddmmmyyy1998 commented 2 years ago

谢谢大佬！