train_yolov8_4detect.py

from ultralytics import YOLO

DATASET = r'coco128.yaml'

if __name__ == '__main__':
    model = YOLO('ultralytics/cfg/models/v8/yolov8n-4detect.yaml')
    model = model.load('yolov8n.pt')
    model.train(data=DATASET, epochs=1000, batch=16)