from ultralytics import YOLO

# Load a model
model = YOLO("yolo26n.pt")  # load a pretrained model (recommended for training)
EPOCHS = 10
IMG_SIZE = 640
BATCH_SIZE = 8
WORKERS = 2
PATIENCE = 5
SEED = 42
# Train the model
results = model.train(data="/Users/nguyenhongvan/recongnize/data/SKU/data_kaggle.yaml", epochs=EPOCHS,
    imgsz=IMG_SIZE,

    batch=BATCH_SIZE,

    workers=WORKERS,

    patience=PATIENCE,

    cache=True,

    amp=True,

    cos_lr=True,

    optimizer="AdamW",

    seed=SEED,

    exist_ok=True
)