YOLOv5模型训练中断,如何恢复训练?断点续传,有效解决。

不知小伙伴在训练自己的YOLOv5数据集时,是否出现训练中断的现象,如何使其恢复原有的数据进度?让我们来看看解决流程。


1.此时数据集正在正常训练,训练100轮次,训练进行中:
YOLOv5模型训练中断,如何恢复训练?断点续传,有效解决。_第1张图片


2.突发电脑重启,再次打开发现根目录下 ‘runs-train-exp7’ 训练了0-2轮次,中断与2轮次:
YOLOv5模型训练中断,如何恢复训练?断点续传,有效解决。_第2张图片

3.此时修改‘–resume’参数下的default值,将False修改为True:

parser.add_argument('--resume', nargs='?', const=True, default=True, help='resume most recent training')

4.YOLOv5自带断点保存,断点续传使其可以恢复训练,如图:
YOLOv5模型训练中断,如何恢复训练?断点续传,有效解决。_第3张图片
YOLOv5模型训练中断,如何恢复训练?断点续传,有效解决。_第4张图片


训练不易,期待大家都能训练出自己理想的模型!
参照博客:https://blog.csdn.net/weixin_45800887/article/details/124222937

你可能感兴趣的:(python,pytorch,深度学习,pycharm,人工智能)