当前位置:   article > 正文

YOLOv8从断点处继续训练_yolov8断点续训

yolov8断点续训

 我租用的服务器做训练。本来想训练120个epoch。训练中间忘记续费了。丫的!服务器供应商那边直接给我关机了,所以在第98个epoch停止了训练。我想从第98个epoch继续训练,需要做下面两步

  1. from ultralytics import YOLO
  2. # (1)加载上次暂停的最后一个epoch保存的模型文件
  3. model = YOLO('./runs/detect/train/weights/last.pt') # load a partially trained model
  4. # (2)重启训练
  5. model.train(resume=True)
  6. # 训练结束评估一下模型
  7. metrics = model.val()

然后就从第98个epoch继续训练了

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/盐析白兔/article/detail/162316
推荐阅读
相关标签
  

闽ICP备14008679号