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