首页 > 其他分享 >yolov5的训练中断恢复

yolov5的训练中断恢复

时间:2023-02-24 19:57:25浏览次数:62  
标签:yolov5 训练 pt 中断 py train weights

Yolov5的恢复训练

使用自带参数-resume

1. train.py文件中找到函数parse_opt,修改参数–resume的默认参数为Ture

           

 

 

2. runs/train/exp*/weights/路径下找到上次训练的最后的保存的权重,加载断点之前最后保留的模型权重last.pt

             

3. 命令行输入命令训练:

!python train.py --weights ./runs/train/exp20/weights/last.pt

参考链接:https://blog.csdn.net/CharmsLUO/article/details/123410081

标签:yolov5,训练,pt,中断,py,train,weights
From: https://www.cnblogs.com/MasJINX17/p/17152935.html

相关文章