Convert onnx to pytorch code.
Project description
onnx-pytorch
Generating pytorch code from ONNX.
Currently support onnx==1.9.0
and torch==1.8.1
.
Usage
from onnx_pytorch import code_gen
code_gen.gen("/path/to/onnx_model", "/path/to/output_dir")
A model.py
file and variables
folder will be created under output_dir
.
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
onnx-pytorch-0.1.1.tar.gz
(32.5 kB
view hashes)
Built Distribution
Close
Hashes for onnx_pytorch-0.1.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | fbedf2ec06f2b3bd448c0ddc81cfd095045da198a098485fea2a55a3ac7e354e |
|
MD5 | 3b6d490d23522a0a0d37bfbc762c166a |
|
BLAKE2b-256 | a4fc8bbbc62698dc045b59e0245a7d7c58322cfc2f907e0fe1b74ae566b35a8b |