YOLO executorch export

yolo_executorch_export.py

from ultralytics import YOLO
import torch

def main():
    model = YOLO("yolov8n.pt") 
    example_input = torch.ones((1, 3, 640, 640))
    exported_program = torch.export.export(model, (example_input,))
    print(exported_program)

if __name__ == '__main__':
    main()