YOLOV8训练中断之后如何继续上次训练?

该方法尝试成功,但是会新创建一个文件夹存放继续训练的权重文件等信息,而且接着训练P R精度并没有接着上次中断的继续更新,这一点有些奇怪还没有找到原因

第一步: 打开ultralytics/yolo/engine/model.py代码

                找到def train() 函数,将self.trainer.model = self.model注释掉

 第二步:打开ultralytics/yolo/engine/trainer.py代码

                首先,找到check_resume()函数, 将resume = self.args.resume注释掉,再让resume = 上次中断生成的last.pt的路径

                 其次,找到resume_training()函数,ckpt =torch.load(上次中断生成的last.pt的路径) 

 切记,下次从头开始训练时,一定要将代码恢复回原样!!!!

该方法YOLOV8不可行,或许YOLOV5可行,YOLOV5可以直接在配置文件中将resume设置为True:Yolov8断点续训/继续训练_q1224352995的博客-CSDN博客

 

  • 6
    点赞
  • 16
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
要在yolov5中继续上次训练,你可以使用参数--resume。这个参数允许你从上次中断的地方继续训练,而不需要重新训练整个模型。具体步骤如下: 1. 首先,你需要找到上次训练停止的检查点文件,通常位于/runs/train/expxxx/weights目录下。这个检查点文件包含了模型在上次训练结束时的权重信息。 2. 接下来,在命令行中运行yolov5的训练命令,并加上参数--resume和检查点文件的路径。例如,如果你的检查点文件是last.pt,那么命令可以是:python train.py --resume weights/last.pt。 3. 运行命令后,yolov5会加载上次训练的权重,并从中断的地方继续训练。你可以继续训练剩下的epoch,或者根据需要选择训练的epoch数量。 通过使用--resume参数,你可以方便地继续上次训练,而不需要重新开始。这对于训练时间较长的模型尤其有用,可以节省训练时间并提高效率。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* [yolov5——断点训练/继续训练【解决方法、使用教程】](https://blog.csdn.net/CharmsLUO/article/details/123410081)[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/m0_67402026/article/details/125346079)[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、付费专栏及课程。

余额充值