剪枝后的模型如何做加速呢？

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

Closed aa12356jm closed 3 years ago

aa12356jm commented 3 years ago

请教一下作者，现在剪枝有结构化剪枝和非结构化剪枝，应该只有结构化剪枝才能实现加速，非结构化剪枝需要硬件的支持，你这里使用的是结构化剪枝吗？结构化剪枝后的模型有把权重为0的filter删除掉吗？谢谢

ZJU-lishuang commented 3 years ago

结构化剪枝，去掉了权重为0的地方，会生成新的cfg文件，也就是一个新的剪枝后的网络