from ultralytics import YOLO
# Load a model
model = YOLO('yolov8n.pt')  # load a pretrained model (recommended for training)
# Train the model
results = model.train(data='data.yaml', epochs=10,  device=0, imgsz=192) 
# export model as TFlite
model.export(format="tflite")