我租用的服务器做训练。本来想训练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继续训练了