资讯 小学 初中 高中 语言 会计职称 学历提升 法考 计算机考试 医护考试 建工考试 教育百科
栏目分类:
子分类:
返回
空麓网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
空麓网 > 计算机考试 > 软件开发 > 后端开发 > Python

yolov5 如何在已经训练完成的模型文件上继续训练

Python 更新时间: 发布时间: 计算机考试归档 最新发布

yolov5 如何在已经训练完成的模型文件上继续训练

继续训练报错:AssertionError: runstrainexpweightslast.pt training to 600 epochs is finished, nothing to resume.

yolov5 如何在已经训练完成的模型文件上继续训练

1.在runstrainexp下的opt.txt中将epochs改为你要训练的总轮数

2.在train.py中将resume参数的default设定为True

3.在yolov5utils下的torch_utils.py中将start_epoch改为上次训练的轮次+1(可以在exp中的result表里看见上次最后训练的轮次,此处轮次指的是标号,自0开始的第一列中最后一行数字)

注意:完全结束过所有epoch后再这样续上训练,前几轮会比较低,但它长得很快不用担心,很快就可以赶回上次训练结束时的程度。以及,别忘了把torch_utils里的那个改回去,不然每次继续训练都从那里开始的!

转载请注明:文章转载自 http://www.konglu.com/
本文地址:http://www.konglu.com/it/1096351.html
免责声明:

我们致力于保护作者版权,注重分享,被刊用文章【yolov5 如何在已经训练完成的模型文件上继续训练】因无法核实真实出处,未能及时与作者取得联系,或有版权异议的,请联系管理员,我们会立即处理,本文部分文字与图片资源来自于网络,转载此文是出于传递更多信息之目的,若有来源标注错误或侵犯了您的合法权益,请立即通知我们,情况属实,我们会第一时间予以删除,并同时向您表示歉意,谢谢!

我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2023 成都空麓科技有限公司

ICP备案号:蜀ICP备2023000828号-2