from ultralytics import YOLOv10

#model = YOLOv10()
# If you want to finetune the model with pretrained weights, you could load the 
# pretrained weights like below
#model = YOLOv10.from_pretrained('jameslahm/yolov10{n/s/m/b/l/x}')
model = YOLOv10.from_pretrained('jameslahm/yolov10n')
# or
# wget https://github.com/THU-MIG/yolov10/releases/download/v1.1/yolov10{n/s/m/b/l/x}.pt
# model = YOLOv10('yolov10{n/s/m/b/l/x}.pt')

model.train(data='/ddn/imu_tsxm1/xm/GPT4V/lung/lung.yaml', epochs=2, batch=64)#, imgsz=640