OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
c062cdbd90
pytorch/caffe2/python/onnx
History
Jeong Ukjae f03156f9df replace blacklist in caffe2/python/onnx/frontend.py (#41777) 
Summary:
Close https://github.com/pytorch/pytorch/issues/41712

Pull Request resolved: https://github.com/pytorch/pytorch/pull/41777

Reviewed By: izdeby

Differential Revision: D22648532

Pulled By: yinghai

fbshipit-source-id: 7f4c9f313e2887e70bb4eb1ab037aea6b549cec7
2020-07-22 10:02:16 -07:00
..
bin remediation of S205607 2020-07-17 17:19:47 -07:00
tests [ONNX] Bump up ONNX submodule to a82c6a7010e2e332d8f74ad5b0c726fd47c85376 (#39372) 2020-06-02 21:08:14 -07:00
__init__.py remediation of S205607 2020-07-17 17:19:47 -07:00
backend_cpp_rep.py
backend_rep.py Add JIT pass to insert permutes for conv ops (#30679) 2019-12-05 18:51:16 -08:00
backend.py Fix typos (#30606) 2019-12-02 20:17:42 -08:00
error.py
frontend.py replace blacklist in caffe2/python/onnx/frontend.py (#41777) 2020-07-22 10:02:16 -07:00
helper.py Add debug logic to c2_ref_test and its helpers (#19359) 2019-04-22 12:08:55 -07:00
onnxifi.py [Onnxifi] Allow adding timeout for OnnxifOp run (#40081) 2020-06-17 16:21:25 -07:00
ONNXOpCoverage.md Fix typos, via a Levenshtein-type corrector (#31523) 2020-01-17 16:03:19 -08:00
README.md
test_onnxifi.py Skip tests if C2/ONNX models cannot be read (#18494) 2019-03-27 11:21:44 -07:00
workspace.py

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