from ultralytics import YOLO

# 1. 모델 로드 (YOLOv8m 사용)
# model = YOLO('yolov8m.pt')
model = YOLO("/home/cuuva/git/Detection_Experiment/fashion_yolo/fashionpedia_exp/yolov8m_fashion_final/weights/best_fashion_16class.pt")

train_results = model.train(
    data="/home/cuuva/git/Detection_Experiment/fashion_yolo/finetuning_exp/finetuning_dataset.yaml",
    epochs=200,            
    imgsz=640,
    batch=-1,              
    device="cuda",
    optimizer='AdamW',
    lr0=0.001,
    patience=50,
    project='finetuned_exp',
    name='yolov8m_finetuned_16class',
)