mirror of https://github.com/zebrajr/pytorch.git synced 2026-01-15 12:15:51 +00:00

Files

Edward Yang d5f8c8f3ba Revert D20121169: [pytorch][PR] ONNX Export Support for CrossEntropyLoss

Test Plan: revert-hammer

Differential Revision:
D20121169

Original commit changeset: 7b56617e8c60

fbshipit-source-id: d7f302d1e54f3c978c3be0a0ad1ee600790a5b27

2020-03-12 20:30:54 -07:00

bin

…

tests

Revert D20121169: [pytorch][PR] ONNX Export Support for CrossEntropyLoss

2020-03-12 20:30:54 -07:00

__init__.py

…

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

…

helper.py

Add debug logic to c2_ref_test and its helpers (#19359 )

2019-04-22 12:08:55 -07:00

onnxifi.py

Combine all the user inputs together and convert them to fp16 (#31898 )

2020-01-08 14:36:42 -08: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