diff --git a/docs/Normal.md b/docs/Normal.md index d69c041..2b654ff 100644 --- a/docs/Normal.md +++ b/docs/Normal.md @@ -12,7 +12,7 @@ from ultralytics import YOLO # Load a model model = YOLO("yolov8n.pt") # load a pretrained model (recommended for training) -success = model.export(format="engine") # export the model to engine format +success = model.export(format="engine", device=0) # export the model to engine format assert success ``` @@ -21,7 +21,7 @@ After executing the above script, you will get an engine named `yolov8n.engine` ### 2. CLI tools ```shell -yolo export model=yolov8n.pt format=engine +yolo export model=yolov8n.pt format=engine device=0 ``` After executing the above command, you will get an engine named `yolov8n.engine` too.