Ultralytics•2mo ago

YOLO Model tune error

I have experienced an issue when model tuning as below. Which I am using Roboflow data set. For coco.yaml this is working fine.

from ultralytics import YOLO
model = YOLO("/kaggle/input/convnext-yolo-100/pytorch/default/1/vit-100e-best.pt")

search_space = {
    "lr0": (1e-5, 1e-1),
}

# Tune hyperparameters on COCO8 for 30 epochs
model.tune(
    data="/kaggle/input/pvelad-2/pvelad-2/data.yaml",
    epochs=30,
    iterations=300,
    optimizer="AdamW",
    space=search_space,
    plots=True,
    save=False,
    val=False
)

from ultralytics import YOLO
model = YOLO("/kaggle/input/convnext-yolo-100/pytorch/default/1/vit-100e-best.pt")

search_space = {
    "lr0": (1e-5, 1e-1),
}

# Tune hyperparameters on COCO8 for 30 epochs
model.tune(
    data="/kaggle/input/pvelad-2/pvelad-2/data.yaml",
    epochs=30,
    iterations=300,
    optimizer="AdamW",
    space=search_space,
    plots=True,
    save=False,
    val=False
)