yolov5 继续训练

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

感谢博主:

yolov5 继续训练_weixin_44057060的博客-优快云博客_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

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值