为什么训练YOLOv8后.pt权重文件大小明显减小？

Question

训练 YOLOv8 模型后，我注意到 .pt 权重文件的大小与其初始大小相比显着减小。 YOLOv8 使用了哪些技术来减少文件大小？此外，与缩小之前的状态相比，这种尺寸减小是否会影响模型的准确性？

Answer 1

这就是所谓的模型剥离。

训练停止后自动完成。

但有时，如果使用 ctrl+c 强制停止训练，则不会执行模型剥离。

在这种情况下，您需要使用此代码手动执行此操作：

from ultralytics.utils.torch_utils import strip_optimizer

strip_optimizer('best.pt')

其背后的具体过程尚不清楚，也许来自 ultralytics 的团队可以更好地解释这一点。