-
Notifications
You must be signed in to change notification settings - Fork 749
Arm backend: Add VIT test #11996
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
Arm backend: Add VIT test #11996
Conversation
Change-Id: Icb1923a9b2cc4740ceff0d4daccba1d12ead6e5f Signed-off-by: Oscar Andersson <[email protected]> Co-authored-by: Tom Allsop <[email protected]>
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/11996
Note: Links to docs will display an error until the docs builds have been completed. ❌ 6 New FailuresAs of commit d250630 with merge base 2aa8084 ( NEW FAILURES - The following jobs have failed:
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
| vit_b_16_model = models.vit_b_16(weights="IMAGENET1K_V1") | ||
| vit = vit_b_16_model.eval() |
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 like this PR hasn't been updated in a while so we're going to go ahead and mark this as |
|
Model is not suitable for testing test/models as it's too big. Closing this. |
Adds unittest for VIT model.
Change-Id: Icb1923a9b2cc4740ceff0d4daccba1d12ead6e5f
cc @digantdesai @freddan80 @per @zingo