Merge pull request dusty-nv#1230 from fivefishstudios/patch-2

Fix command line for onnx_export.py
shoaib42 · Sep 20, 2021 · 9a94ef5 · 9a94ef5
2 parents cb73202 + a6eca8c
commit 9a94ef5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/pytorch-plants.md b/docs/pytorch-plants.md
@@ -106,7 +106,7 @@ Note that the models are saved under `jetson-inference/python/training/classific
 Just like with the Cat/Dog example, next we need to convert our trained model from PyTorch to ONNX, so that we can load it with TensorRT:
 
 ``` bash
-python3 onnx_export.py --model-dir=model/plants
+python3 onnx_export.py --model-dir=models/plants
 ```
 
 This will create a model called `resnet18.onnx` under `jetson-inference/python/training/classification/models/plants/`