-
Notifications
You must be signed in to change notification settings - Fork 2.7k
add dynamic export and visualize to pytorch2onnx #463
New issue
Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? # to your account
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.