YOLOv5继续训练【断点续传】

项目场景:

YOLOv5训练了很久突然发现中断了心态爆炸,重新训练浪费时间,万一再中断呢。
想起来之前YOLOv3可以根据log继续训练,所以YOLOv5是不似乎也可以呢!
抱着试一试的想法去csdn和百度上翻了半天都是YOLOv5训练数据的保姆级教程,心塞塞…
不过功夫不负有心人 终于解决了!Mark一下。

问题描述:

YOLOv5训练500epoch,速度异常缓慢,30个小时才不到200次还中断了。。。
如何能够断点续传呢?

解决方案:

Pytorch里有一个resume功能,YOLOv5是基于Pytorch的,应该也有的
果然!

然后发现default是默认False的,稍微修改一下,断点续传成果!


原创链接:https://blog.csdn.net/weixin_44368687/article/details/120395258

你可能感兴趣的:(机器学习,目标检测,人工智能,计算机视觉)