Fix Deprecation Warnings torch.GradScaler and torch.autocast#930

Open

IliasChair wants to merge 2 commits intoFAIR-Chem:mainfrom IliasChair:fix-deprecation-warnings

+18-18

Commits on Nov 30, 2024

- replace all occurrences of torch.cuda.amp.autocast(args...) with torch.autocast("cuda", args...)
IliasChair14
committed

Commits on Dec 11, 2024

Merge branch 'main' into fix-deprecation-warnings
IliasChair
authored