Fix command line for onnx_export.py

Fix command line for onnx_export.py command. --model-dir should be 'models/plants', and not 'model/plants'
shoaib42 · Sep 20, 2021 · a6eca8c · a6eca8c
1 parent cb73202
commit a6eca8c
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/`