1、train.py中加上resume=True
2、ultralytics/engine/trainer.py
注释# resume = self.args.resume
加上resume = '/ultralytics/runs/detect/train4/weights/last.pt'
def check_resume(self, overrides):
"""Check if resume checkpoint exists and update arguments accordingly."""
# resume = self.args.resume
resume = '/ultralytics/runs/detect/train4/weights/last.pt'
3、ultralytics/engine/trainer.py
加上ckpt = torch.load("/ultralytics/runs/detect/train4/weights/last.ptlast.pt")
def resume_training(self, ckpt):
"""Resume YOLO training from given epoch and best fitness."""
ckpt = torch.load('/ultralytics/runs/detect/train4/weights/last.pt')