Skip to content

Add cuda resnet50 test to smoke test #7020

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

Merged
merged 3 commits into from
Dec 8, 2022
Merged

Conversation

atalman
Copy link
Contributor

@atalman atalman commented Dec 8, 2022

Execute resnet50 classify test on the cuda device

cc @pmeier

Copy link
Contributor

@datumbox datumbox left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, thanks!

@atalman atalman merged commit 09030c0 into pytorch:main Dec 8, 2022
facebook-github-bot pushed a commit that referenced this pull request Dec 16, 2022
Summary:
* Add cuda resnet50 test

* Fix path

* Tune vision smoke test

Reviewed By: YosuaMichael

Differential Revision: D42046599

fbshipit-source-id: d5aaa4dcd7a07730347d3417f3f94eed70d5a91c
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants