-
Notifications
You must be signed in to change notification settings - Fork 1.1k
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 #277
Conversation
Hi do you want to write a doc on how to use this tool? Add contents to here https://mmediting.readthedocs.io/en/latest/tools_scripts.html?highlight=onnx#convert-to-onnx-experimental . |
Codecov Report
@@ Coverage Diff @@
## master #277 +/- ##
==========================================
+ Coverage 79.94% 79.99% +0.05%
==========================================
Files 159 160 +1
Lines 7942 8003 +61
Branches 1177 1185 +8
==========================================
+ Hits 6349 6402 +53
- Misses 1449 1456 +7
- Partials 144 145 +1
Flags with carried forward coverage won't be shown. Click here to find out more.
Continue to review full report at Codecov.
|
docs/tools_scripts.md
Outdated
|
||
### Evaluate ONNX model with ONNXRuntime | ||
|
||
We provide `tools/ort_test.py` to evaluate ONNX model with ONNXRuntime backend. |
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.
ort is cryptic. How about renaming the file to onnx_runtime_test.py
?
Close this PR. |
Hi
This PR adds an ONNXRuntime test tool to align results between PyTorch and ONNXRuntime.
Tested on mattor and restorer models.