-
Notifications
You must be signed in to change notification settings - Fork 7k
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
port prototype tests to new utilities #8022
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/vision/8022
Note: Links to docs will display an error until the docs builds have been completed. ✅ You can merge normally! (4 Unrelated Failures)As of commit cb5c813 with merge base 67f3ce2 (): FLAKY - The following jobs failed but were likely due to flakiness present on trunk:
BROKEN TRUNK - The following jobs failed but were present on the merge base:👉 Rebase onto the `viable/strict` branch to avoid these failures
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
], | ||
) | ||
|
||
def make_label(*, extra_dims=(), categories=10, dtype=torch.int64, device="cpu"): |
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've opted to move this here from the previous prototype_common_utils
module, since this module was the only one that imported from there. Plus, we had unused stuff in prototype_common_utils
.
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.
Thanks @pmeier
Reviewed By: vmoens Differential Revision: D50789081 fbshipit-source-id: 8a96e3a69c65b77f72172828516cdfc8b4ac3e74
Follow-up to #8013 (comment). The prototype tests are not legacy as they will not be removed. This PR migrates the used utilities to the current ones.
cc @vfdev-5 @bjuncek