yolov5prune icon indicating copy to clipboard operation
yolov5prune copied to clipboard

Published 20 hours ago • verified publisher icon midasklr

请问博主执行了pruned.py之后生成了一个orign_model.pt和一个pruned_model.pt，可是为什么orign_model.pt比我剪枝前的模型大很多，而且精度不但比剪枝前的模型精度低，还比生成的pruned_model.pt精度低

Open wyzwyzzzzz opened this issue 3 years ago • 2 comments

wyzwyzzzzz avatar Nov 30 '21 08:11 wyzwyzzzzz

博主已经在博客帮忙回答了，感谢！ 以下是博主的原回答： 剪枝前的模型是半精度存储的（可以看看模型save的代码部分），所以orign_model.pt模型大概是他的两倍；精度的问题不太好说，情况可能很多，但是正常来说应该不会出现这种情况。

wyzwyzzzzz avatar Dec 03 '21 01:12 wyzwyzzzzz

博主已经在博客帮忙回答了，感谢！ 以下是博主的原回答： 剪枝前的模型是半精度存储的（可以看看模型save的代码部分），所以orign_model.pt模型大概是他的两倍；精度的问题不太好说，情况可能很多，但是正常来说应该不会出现这种情况。 请问应该怎么以半精度存储orign_model.pt呢？如果以半精度存储orign_model.pt的话，它应该在14MB左右吧

Nwaf-sl avatar May 10 '22 10:05 Nwaf-sl