Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- THIS IS THE COMMAND YOU EXECUTE TO START TRAINING THE MODEL, READ NOTE BELOW TO SEE INFO ON EPOCHS:
- yolo task=detect mode=train imgsz=640 data=data.yaml epochs=100 batch=16 name=putyourmodelnamehere
- USE THIS AFTER YOU'VE FINISHED TRAINING TO CONVERT YOUR MODEL TO ONNX SO YOU CAN USE IT WITH AIMMY:
- yolo export model=best.pt format=onnx
- NOTE:
- use 1 - 1000 epochs (the more epochs the longer the training but better end results for model)
- YOU NEED THIS:
- pip install ultralytics
- YOU ONLY NEED THIS IF THE ABOVE IS NOT WORKING (get requirements.txt from:https://docs.ultralytics.com/quickstart/)
- pip install -r requirements.txt
- YOU NEED THIS:
- https://pytorch.org/get-started/locally/
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement