yolonew.py

from ultralytics import YOLO

if __name__ == '__main__':

    print("Attempting to load YOLO26 Medium...")
    model = YOLO("yolo26m-seg.pt") 

    model.train(
        data=r"C:\Users\charu\Desktop\My_Project\teacher_v2_dataset\teacher_v2.yaml",
        epochs=200,
        imgsz=640,
        
 
        batch=4,       
        
        device=0,
        name="teacher_v2_medium_26"
    )