-
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
[Feature] add onnxruntime test tool #498
Conversation
Codecov Report
@@ Coverage Diff @@
## master #498 +/- ##
==========================================
+ Coverage 86.48% 86.60% +0.11%
==========================================
Files 97 99 +2
Lines 4974 5160 +186
Branches 807 834 +27
==========================================
+ Hits 4302 4469 +167
- Misses 519 533 +14
- Partials 153 158 +5
Flags with carried forward coverage won't be shown. Click here to find out more.
Continue to review full report at Codecov.
|
LGTM. Ping @RunningLeon if there is any comments. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
* add onnxruntime test tool, update pytorch2onnx to support slice export * onnx convert with custom output shape, update test code * update pytorch2onnx, add rescale_shape support, add document * update doc for lint error fixing * remove cpu flag in ort_test.py * change class name, fix cuda error * remote comment * fix bug of torch2onnx * mIOU to mIoU
Fix `disable_attention_slicing` in pipelines.
Hi
This PR adds an ONNXRuntime test tool to align results between PyTorch and ONNXRuntime.
Pytorch2onnx is also updated to support rescale output shape (a large output shape might cause OOM).
TODO