Torch-Pruning
Torch-Pruning copied to clipboard
剪枝之后准确率变化
你好,感谢你的工作,我使用yolov4在一个数据集上进行训练,map为64%,使用该权重进行剪枝后,模型大小由256MB减小到了147MB,速度也有很大提升,但是剪枝得到的模型几乎不能检测出任何东西,后续进行微调之后,准确率只有43%,这是不太正常的,剪枝之前是要对模型稀疏化训练吗,我看example中yolov5的代码好像没有稀疏训练的过程。
@neuhsm,您好,这个库目前并没有实现太多算法,只是提供了最基础的裁剪功能。微调阶段学习率建议降低一些,可以缩小到原先的0.1。
请问这个问题后续有更好的方法吗