项目场景:
YOLOv5训练了很久突然发现中断了心态爆炸,重新训练浪费时间,万一再中断呢。
想起来之前YOLOv3可以根据log继续训练,所以YOLOv5是不似乎也可以呢!
抱着试一试的想法去csdn和百度上翻了半天都是YOLOv5训练数据的保姆级教程,心塞塞…
不过功夫不负有心人 终于解决了!Mark一下。
问题描述:
YOLOv5训练500epoch,速度异常缓慢,30个小时才不到200次还中断了。。。
如何能够断点续传呢?
解决方案:
Pytorch里有一个resume功能,YOLOv5是基于Pytorch的,应该也有的
果然!
然后发现default是默认False的,稍微修改一下,断点续传成果!
原创链接:https://blog.youkuaiyun.com/weixin_44368687/article/details/120395258