frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Updated PyTorch ONNX exporter (markdown)

Justin Chu
2022-05-25 09:53:57 -07:00
parent 441f0245a7
commit 19ab05ac9d
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
PyTorch-ONNX-exporter.md

@ -266,6 +266,12 @@ You can use `pytest` to run tests in parallel and generate a coverage report.
python -m pytest -n auto --cov --cov-report "xml:test/coverage.xml" test/onnx/test_pytorch_onnx_onnxruntime.py
```
To test everything minus caffe2 and gpu tests, use
```sh
python -m pytest -n auto --cov --cov-report "xml:test/coverage.xml" --ignore "test/onnx/test_pytorch_onnx_caffe2.py" --ignore "test/onnx/test_pytorch_onnx_caffe2_quantized.py" --ignore "test/onnx/test_caffe2_common.py" --ignore "test/onnx/test_pytorch_onnx_onnxruntime_cuda.py"
```
# Links
- [User-facing docs](https://pytorch.org/docs/master/onnx.html).