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

  • 1
    点赞
  • 19
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
根据引用的内容,如果想要继续训练yolov5模型,有几种情况需要考虑。首先,如果训练过程中中断了,可以继续训练。其次,如果训练完了但是未收敛,可以在原有的基础上继续训练。最后,如果训练完原有epoch后,想要继续训练更多的epoch,也是可以的。 总的来说,无论是训练过程中中断了还是训练完了但是未收敛,都可以使用之前训练得到的权重和学习率等参数继续训练。这样可以在之前的基础上继续优化模型,提高模型的准确性。 例如,如果训练完了30个epoch但是未收敛,想要继续训练40个epoch,总共70个epoch,可以直接使用之前训练了30个epoch的权重进行继续训练。这样可以在之前训练得到的基础上进一步提升模型的性能。 需要注意的是,继续训练时需要合理设置学习率和其他超参数,以确保训练的稳定性和效果。此外,继续训练时还可以考虑使用一些技巧,如学习率衰减、增加数据增强等,来进一步提升模型的性能。 综上所述,可以通过继续训练来优化yolov5模型,无论是在训练过程中中断了还是训练完了但是未收敛,都可以使用之前的权重和参数进行继续训练,以提升模型的准确性和性能。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* [yolov5 继续训练](https://blog.csdn.net/weixin_44057060/article/details/124349715)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] - *2* *3* [YOLOV5-断点训练/继续训练](https://blog.csdn.net/weixin_42886071/article/details/126676646)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值