-
Notifications
You must be signed in to change notification settings - Fork 2.6k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
add dynamic export and visualize to pytorch2onnx #463
Conversation
We may also add usage here |
Codecov Report
@@ Coverage Diff @@
## master #463 +/- ##
==========================================
- Coverage 86.52% 86.50% -0.02%
==========================================
Files 97 97
Lines 4964 4966 +2
Branches 807 806 -1
==========================================
+ Hits 4295 4296 +1
- Misses 515 517 +2
+ Partials 154 153 -1
Flags with carried forward coverage won't be shown. Click here to find out more.
Continue to review full report at Codecov.
|
Hi @grimoire |
@xvjiarui done. |
Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>
* add dynamic export and visualize to pytorch2onnx * update document * fix lint * fix dynamic error and add visualization * fix lint * update docstring * update doc * Update help info for --show Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com> * fix lint Co-authored-by: maningsheng <maningsheng@sensetime.com> Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>
* fix_merge_config * pre-commit * add changelog
Hi
This PR adds dynamic export and visualize of pytorch2onnx.
Tested on FCN, PSPNet, DeeplabV3, DeeplabV3+.
Note that dynamic shape support of PSPNet is not fully supported cause ONNX does not support dynamic export of AdaptiveAvgPool.