增加CBAM,C3STR后如何剪枝

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

Open ruoyuchen0903 opened 1 year ago

ruoyuchen0903 commented 1 year ago

想问问，自定义改变网络中的模块，如增加注意力机制或者融入transformer后怎么剪枝？

ZJU-lishuang commented 1 year ago

parse_module_defs2 该函数选择剪枝模块