How to resume/continue from already trained YOLO5 Training Epoch?

Question

I have started the Yolo5 Training with custom data

The command I have used:

!python train.py --img 640 --batch-size 32 --epochs 5 --data /content/drive/MyDrive/yolov5_dataset/dataset_Trafic/data.yaml --cfg /content/drive/MyDrive/yolov5/models/yolov5s.yaml  --name Model

Training started as below & completed:

For resuming/continue for more epoch I have below command

!python train.py --img 640 --batch-size 32 --epochs 6 --data /content/drive/MyDrive/yolov5_dataset/dataset_Trafic/data.yaml --weights /content/drive/MyDrive/yolov5/runs/train/Model/weights/best.pt --cache --exist-ok

But still the Training start from the scratch. How to continue from previous epoch.

Also I tried with resume command

!python train.py --epochs 10 --resume

but I am getting below error message

Answer 1

I'm not sure if this would work for YOLOv5 but this is how to resume training in YOLOv8 from the documentation:

Python

from ultralytics import YOLO
model = YOLO('path/to/last.pt')  # load a partially trained model
results = model.train(resume=True)

CLI

yolo train resume model=path/to/last.pt

Answer 2

Open runs/train/Model/opt.yaml file and change epochs no. try again

!python train.py  --weights /content/drive/MyDrive/yolov5/runs/train/Model/weights/last.pt  --resume

Answer 3

Try this: In the runs folder(where your previous weights are stored), find a file with name "args.yaml" inside the file, augment the epochs count by the no. of epochs you want to train your model now. For example: If previously your model was trained for 100 epochs then, the args.yaml file would have an attribute: "epochs: 100" Lets say you was to continue training for 50 more epochs then change epochs to 150 i.e. "epochs: 150" save the file and then execute training command.