YOLOv8从断点处继续训练

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

from ultralytics import YOLO 
# (1)加载上次暂停的最后一个epoch保存的模型文件
model = YOLO('./runs/detect/train/weights/last.pt')  # load a partially trained model

# (2)重启训练
model.train(resume=True)

# 训练结束评估一下模型
metrics = model.val() 

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

猜你喜欢

转载自blog.csdn.net/Albert233333/article/details/131388478