fix deprecated torch.cuda.amp.GradScaler
FutureWarning for pytorch 2.4+
#5249
Job | Run time |
---|---|
1m 20s | |
1m 20s |
torch.cuda.amp.GradScaler
FutureWarning for pytorch 2.4+
#5249
Job | Run time |
---|---|
1m 20s | |
1m 20s |