-
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
[proto][ci] Try add GPU ci for prototype transforms #6919
Conversation
38f2e17
to
d826dd0
Compare
d826dd0
to
ad9ee3a
Compare
@osalpekar I made updates you mentioned in #6957 (comment) and seems like this solves CUDA problem. Please review this PR and we can merge it to enable CUDA tests. |
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.
Awesome!
Hey @vfdev-5! You merged this PR, but no labels were added. The list of valid labels is available at https://github.com/pytorch/vision/blob/main/.github/process_commit.py |
Summary: * Try add GPU ci for prototype transforms * Update prototype-transforms-tests-linux-gpu.yml * Update prototype-transforms-tests-linux-gpu.yml * Update prototype-transforms-tests-linux-gpu.yml * Update prototype-transforms-tests-linux-gpu.yml * Update prototype-transforms-tests-linux-gpu.yml * add small tolerance for adjust_contrast on CUDA * [skip-ci] update Reviewed By: YosuaMichael Differential Revision: D42046597 fbshipit-source-id: a90c74df96d581f510987aaddb419c3a0687e203 Co-authored-by: Philip Meier <github.pmeier@posteo.de>
Another attempt to enable tests on GPU for prototype transforms, based on https://github.com/pytorch/vision/blob/main/.github/workflows/test-linux-gpu.yml
Previous attempt: #6665
cc @pmeier @seemethere @bjuncek