赞
踩
我租用的服务器做训练。本来想训练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继续训练了
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。