YOLOV8的模型为什么在最后会减小?本来还是二十多兆,训练结束就成了6兆,是在哪个函数减小的啊?YOLOV5也会吗?
2条回答 默认 最新
- 爱晚乏客游 2023-11-21 17:52关注
都会,训练结束之后会将模型里面的额外的一些参数释放掉,比如优化器之类的,具体存了哪些信息你可以看代码,也可以修改你要存的东西。
最后这里的strip_optimizer就是训练结束之后,将ckpt里面无关的东西都去掉,只留下模型本回答被题主选为最佳回答 , 对您是否有帮助呢?解决 无用评论 打赏 举报