Replace torch.cuda.amp.autocast with torch.amp.autocast('cuda',...)
#73
Annotations
1 warning
|
The following actions use a deprecated Node.js version and will be forced to run on node20: EndBug/[email protected]. For more info: https://github.blog/changelog/2024-03-07-github-actions-all-actions-will-run-on-node20-instead-of-node16-by-default/
|
Loading