YOLOv5继续训练【断点续传】

项目场景:

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


问题描述:

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

解决方案:

Pytorch里有一个resume功能,YOLOv5是基于Pytorch的,应该也有的
果然!
YOLOv5继续训练【断点续传】_第1张图片
然后发现default是默认False的,稍微修改一下,断点续传成果!
开开心心继续训练去!

你可能感兴趣的:(YOLOv5,pytorch,深度学习,python)