-
Notifications
You must be signed in to change notification settings - Fork 751
NXP backend: Add MobileNetV2 example model and test #12892
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
NXP backend: Add MobileNetV2 example model and test #12892
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/12892
Note: Links to docs will display an error until the docs builds have been completed. ❌ 1 New Failure, 1 PendingAs of commit 39c59c8 with merge base 49bc664 ( NEW FAILURE - The following job has failed:
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
|
@pytorchbot label "release notes: nxp" |
c2dae16 to
afb0257
Compare
robert-kalmar
left a comment
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.
Looks good to me. The failing tests seems unrelated to this PR.
digantdesai
left a comment
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.
I know this is just AOT but if you ran the PTE internally on your end what's the error looks like when you compare exported, quantized GraphModule on PyTorch vs. Neutron Simulator elementwise? i.e. ATOL, RTOL for tensor.all_close?
afb0257 to
670002b
Compare
We calculate and check the MSE, and the threshold is set to 8.6e-2 (note the output is FP32) Here is the result from CPU and NPU run: CPU: Neutron NPU |
670002b to
39c59c8
Compare
### Summary Add MobileNetV2 model as example and for integration testing ### Test plan Support for testing full conversion on this model is included in `run_aot_example.sh`. --------- Co-authored-by: Lukas Sztefek <[email protected]>
Summary
Add MobileNetV2 model as example and for integration testing
Test plan
Support for testing full conversion on this model is included in
run_aot_example.sh.cc @skywall @robert-kalmar