train.py

# %pip install comet_ml --quiet

import comet_ml
comet_ml.login(project_name='final_midwater_supercategory_detector_v2')
#comet_ml.start(mode="get", experiment_key="27f573d41b4c49068270f1d3fb6be523")
import torch
print("CUDA Available:", torch.cuda.is_available())
if torch.cuda.is_available():
        print("GPU Name:", torch.cuda.get_device_name(0))


from ultralytics import YOLO

model = YOLO("yolo11x.yaml")
model = YOLO('yolo11x.pt')  # Load a pretrained model
model = YOLO("yolo11x.yaml").load("yolo11x.pt")

results = model.train(data='/data/james/final_split/midwater/midwater_supercategory_detector.yaml', epochs=100, imgsz=640, patience=5, val=True, device=[-1,-1])