yolov5 继续训练

昨天我训练了300轮,发现模型未收敛,想再继续训练200.采用博主博客中方法2,成功跑起来了

感谢博主:

yolov5 继续训练_weixin_44057060的博客-CSDN博客_yolov5 继续训练

具体做法是:
 

不是直接在train.py设置weight参数为训练好的权重(last.pt),然后设置epoch为300,这样不行,因为学习率等超参数启动时还是按照默认的。。。(自己把一些超参数设置好也可以,但感觉有点烦)
1. 在train.py代码里将 epoch 改为500
2. 在train.py设置resume为True
3. 在train.py大概190行如下位置之前加上代码

  ckpt['epoch'] = 300

yolov5 继续训练_第1张图片

你可能感兴趣的:(机器学习,学习,深度学习,机器学习,yolov5)