OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
f587267dc7
pytorch/caffe2/python/onnx
History
Lu Fang a6eec0c60f Upgrade onnx submodule to 85546f8c44e627f8ff1181725d03cc49f675e44f (#66427) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/66427

Update the onnx submodule, so https://github.com/pytorch/pytorch/pull/66140 can land.

Test Plan: ci

Reviewed By: ezyang

Differential Revision: D31544610

fbshipit-source-id: 94831ef531bbd654a6aeb744cd53a38155848079
2021-10-12 09:46:08 -07:00
..
bin Drop unused imports from caffe2/python (#49980) 2021-01-05 13:17:46 -08:00
tests Upgrade onnx submodule to 85546f8c44e627f8ff1181725d03cc49f675e44f (#66427) 2021-10-12 09:46:08 -07:00
__init__.py remediation of S205607 2020-07-17 17:19:47 -07:00
backend_cpp_rep.py Lint trailing newlines (#54737) 2021-03-30 13:09:52 -07:00
backend_rep.py Remove __future__ imports for legacy Python2 supports (#45033) 2020-09-23 17:57:02 -07:00
backend.py Fix ONNX forward compatibility (#59327) 2021-06-02 12:39:56 -07:00
error.py Remove __future__ imports for legacy Python2 supports (#45033) 2020-09-23 17:57:02 -07:00
frontend.py Remove redundant code for unsupported Python versions (#49486) 2021-01-06 12:45:46 -08:00
helper.py Drop unused imports from caffe2/python (#49980) 2021-01-05 13:17:46 -08:00
onnxifi.py Add onnxifi interface for set/get options (#52388) 2021-02-18 20:12:34 -08:00
ONNXOpCoverage.md Fix typos, via a Levenshtein-type corrector (#31523) 2020-01-17 16:03:19 -08:00
README.md Lint trailing newlines (#54737) 2021-03-30 13:09:52 -07:00
test_onnxifi.py Lint trailing newlines (#54737) 2021-03-30 13:09:52 -07:00
workspace.py Remove __future__ imports for legacy Python2 supports (#45033) 2020-09-23 17:57:02 -07:00

README.md

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/

License

MIT License