3 Commits

Author SHA1 Message Date
QingChuanWS
2bd28435cf
Add simplify option in pytorch2onnx.py and Update doc (#200)
* add simplify option in pytorch2onnx.py and update relevant doc

* update doc about pytorch2onnx

* update code

* update docs
2021-04-09 13:05:25 +08:00
QingChuanWS
8eb845e718
support dynamic shape export to onnx (#175)
* support dynamic shape export to onnx

* update the code and reset the simple_test() of image.py

* fix dynamic_axes name is not a valid input/output name
2021-03-17 15:50:03 +08:00
Lei Yang
edffbcc486
Add pytorch2onnx (#20)
* add pytorch2onnx

* add pytorch2onnx API
2020-09-30 20:13:36 +08:00