torch.cuda.amp.autocast()
-> torch.amp.autocast("cuda")
#1921
Merged
qgallouedec merged 2 commits intomainfrom fix-amp-context-managerAug 12, 2024
+37-33
torch.cuda.amp.autocast()
-> torch.amp.autocast("cuda")
#1921