-
-
Notifications
You must be signed in to change notification settings - Fork 624
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
Fix deprecated statement #3307
Fix deprecated statement #3307
Conversation
Replaced torch.cuda.amp.autocast with torch.amp.autocast("cuda",...).
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 for the PR @Gulin7
I left few comments to address
@vfdev-5 I addressed your comments, tag me and let me know if there is anything else that needs improvement. |
Some tests are failing as well: https://github.com/pytorch/ignite/actions/runs/12045408777/job/33647387903 |
@@ -1,31 +1,4 @@ | |||
{ | |||
"nbformat": 4, |
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.
Can you please make just necessary changes without all unrelated updates?
@Gulin7 there is a failing test with error message:
and also there is this comment to address: #3307 (comment) Can you please fix them? Thanks! |
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.
LGTM
Fixes #3304
Description:
Replaced all occurances of torch.cuda.amp.autocast with torch.amp.autocast('cuda',...)