saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-15 21:00:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/caffe2/python/onnx
History
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.py Fix typos (#30606) 2019-12-02 20:17:42 -08: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
error.py
frontend.py
helper.py
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
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