Yolov5意外中断重新训练(自用)

本文介绍了如何在Yolov5训练过程中处理意外中断,包括设置`resume`为默认值,修改`smart_resume`函数以正确恢复训练,确保`epoch`设置,并提供重新训练命令。
部署运行你感兴趣的模型镜像

Yolov5训练意外中断或想继续训练,

解决方法:

1.保证runs目录下的exp是最大的
2.在train.py中找到resume,将default改为True

在这里插入图片描述

3.在train.py中找到smart_resume函数,ctrl到torch_utils.py,进行如下修改
ckpt['epoch']=下一轮训练的编号
start_epoch = ckpt['epoch'] #+1去掉

在这里插入图片描述

4.重新训练
python train.py --weights last.pt路径

在这里插入图片描述

您可能感兴趣的与本文相关的镜像

Yolo-v5

Yolo-v5

Yolo

YOLO(You Only Look Once)是一种流行的物体检测和图像分割模型,由华盛顿大学的Joseph Redmon 和Ali Farhadi 开发。 YOLO 于2015 年推出,因其高速和高精度而广受欢迎

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值