Train the selected model structure for your dataset

[smallMantou]

Search before asking

• I have searched the YOLOv8 issues and discussions and found no similar questions.

Question

Hello, I have some questions about training my own dataset. I want to start training my dataset from scratch. How do I know if my model is based on n, s, m, or l? How can I make changes to train S? Does the following image indicate that I am based on n

Additional

No response

[deepukr007]

I think AMP check always happens on yolov8n, I see the same thing everytime.. You can ignore it.
If your YAML file has name such as yolov8s_{....} .yaml your model will be based on yolov8s..
You can also cross check this by looking at the total model parameters .

[smallMantou]

How can I change it to be based on YOLOv8S?

[glenn-jocher]

Hello! To switch your model base to YOLOv8s, you'll need to use a YOLOv8s model configuration file. You can specify this in your training command by replacing the model argument with the path to a YOLOv8s YAML file. For example:

yolo detect train data=your_dataset.yaml model=yolov8s.yaml epochs=100 imgsz=640

Make sure you have the yolov8s.yaml file in your directory, or you can download it from the Ultralytics repository if it's not already included. This will set up your training to use the YOLOv8s architecture. Happy training! 🚀