yolov5_prune icon indicating copy to clipboard operation
yolov5_prune copied to clipboard

Published 20 hours ago • verified publisher icon ZJU-lishuang

增加CBAM,C3STR后如何剪枝

Open ruoyuchen0903 opened this issue 1 year ago • 1 comments

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

ruoyuchen0903 avatar Apr 25 '23 07:04 ruoyuchen0903

parse_module_defs2 该函数选择剪枝模块

ZJU-lishuang avatar Apr 26 '23 03:04 ZJU-lishuang