Skip to content

Commit

Permalink
Fix MMDetection model to ONNX command
Browse files Browse the repository at this point in the history
  • Loading branch information
Rishit-dagli committed Nov 21, 2021
1 parent a7a16af commit 99b0779
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/useful_tools.md
Original file line number Diff line number Diff line change
Expand Up @@ -310,7 +310,7 @@ Params: 37.74 M
We provide a script to convert model to [ONNX](https://github.com/onnx/onnx) format. We also support comparing the output results between Pytorch and ONNX model for verification.

```shell
python tools/deployment/pytorch2onnx.py ${CONFIG_FILE} ${CHECKPOINT_FILE} --output_file ${ONNX_FILE} [--shape ${INPUT_SHAPE} --verify]
python tools/deployment/pytorch2onnx.py ${CONFIG_FILE} ${CHECKPOINT_FILE} --output-file ${ONNX_FILE} [--shape ${INPUT_SHAPE} --verify]
```

**Note**: This tool is still experimental. Some customized operators are not supported for now. For a detailed description of the usage and the list of supported models, please refer to [pytorch2onnx](tutorials/pytorch2onnx.md).
Expand Down

0 comments on commit 99b0779

Please sign in to comment.