mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
796a39ba85
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/22981 Previous import was 806aa863020fa180e57f576cb032ec44ce8ddcca Included changes: - **[70706498](https://github.com/onnx/onnx/commit/70706498)**: TensorProto::INT8 & INT16 were missed here (#2164) <ZINEKS> - **[8218a4ea](https://github.com/onnx/onnx/commit/8218a4ea)**: Fix LabelEncoder's shape inference (#2170) <Wei-Sheng Chin> - **[0f1a9a1c](https://github.com/onnx/onnx/commit/0f1a9a1c)**: Fixing a unit test in Cumsum Operator (#2157) <Jeff Saremi> - **[2c03cff0](https://github.com/onnx/onnx/commit/2c03cff0)**: [New Operator] CumSum (#2030) <Jeff Saremi> - **[220b8300](https://github.com/onnx/onnx/commit/220b8300)**: Fix globalpool output shape (#2147) <daquexian> Reviewed By: benoitsteiner Differential Revision: D16341736 fbshipit-source-id: 7e7a2684d8c821991231bfd6558f9f6cb4fb05fb |
||
|---|---|---|
| .. | ||
| bin | ||
| tests | ||
| __init__.py | ||
| backend.py | ||
| backend_cpp_rep.py | ||
| backend_rep.py | ||
| error.py | ||
| frontend.py | ||
| helper.py | ||
| onnxifi.py | ||
| ONNXOpCoverage.md | ||
| README.md | ||
| test_onnxifi.py | ||
| workspace.py | ||
Caffe2 implementation of Open Neural Network Exchange (ONNX)
Usage
Installation
onnx-caffe2 is installed as a part of Caffe2. Please follow the instructions to install Caffe2.
Folder Structure
- ./: the main folder that all code lies under
- frontend.py: translate from caffe2 model to onnx model
- backend.py: execution engine that runs onnx on caffe2
- tests/: test files
Testing
onnx-caffe2 uses pytest as test driver. In order to run tests, first you need to install pytest:
pip install pytest-cov
After installing pytest, do
pytest
to run tests.
Testing coverage issues/status: https://github.com/caffe2/caffe2/blob/master/caffe2/python/onnx/ONNXOpCoverage.md
Development
During development it's convenient to install caffe2 in development mode:
cd /path/to/caffe2
pip install -e caffe2/