train.py

import torch
torch.cuda.is_available()
from utils.train_speaker import train_speaker
from ultralytics import YOLO

def train(config):

    model = YOLO("yolo11m.pt")

    model.info()

    model.train(data='data.yaml',epochs=10, device='cuda', imgsz=640, batch=16)

    
    train_speaker(config)