hawcat
文章20
标签28
分类8

文章分类

文章归档

YOLOv5 - 恢复意外停止的训练

YOLOv5 - 恢复意外停止的训练

场景:原本还在训练,设定200轮,跑到180多轮手贱打开Visual Studio结果内存溢出报错了

后来发现Yolo自带有恢复之前训练的参数,对我这个善始善终有强迫症的人来说简直是福音,(虽然我这个小模型在前100轮就已经有了很好的回归率表现了)

QQ图片20230316210801.png
意外暂停


QQ图片20230316210805.png
设置恢复参数为True


QQ图片20230316210809.png
从上次停止的exp继续训练 而不是新建exp

:D 获取中...

本文作者:hawcat
本文链接:https://hawcat.cn/2023/03/16/yolo/
版权声明:本文采用 CC BY-NC-SA 3.0 CN 协议进行许可
×