yolov5prune icon indicating copy to clipboard operation
yolov5prune copied to clipboard

请问在稀疏训练以及微调之后参数量跟计算量会有何改变

Open suhanling opened this issue 3 years ago • 1 comments

如题,因为大佬的这两个代码取消了fuze ,跑完之后看不到这两个参数,不知道这两个会不会有变化

suhanling avatar Nov 24 '21 12:11 suhanling

如题,因为大佬的这两个代码取消了fuze ,跑完之后看不到这两个参数,不知道这两个会不会有变化

取消了fuse保留bn层方便根据bn层weights剪枝,你可以试试在剪枝微调之后,再对conv和bn进行融合,统计参数量和计算量。

midasklr avatar Nov 26 '21 01:11 midasklr