change torch.cuda.amp.GradScaler to torch.GradScaler("cuda") #3901
Job | Run time |
---|---|
15m 24s | |
53m 1s | |
22m 6s | |
30m 18s | |
22m 54s | |
20m 13s | |
23m 7s | |
17m 15s | |
24m 25s | |
19m 46s | |
18m 48s | |
34m 59s | |
34m 13s | |
49m 39s | |
30m 25s | |
22m 55s | |
7h 19m 28s |
Job | Run time |
---|---|
15m 24s | |
53m 1s | |
22m 6s | |
30m 18s | |
22m 54s | |
20m 13s | |
23m 7s | |
17m 15s | |
24m 25s | |
19m 46s | |
18m 48s | |
34m 59s | |
34m 13s | |
49m 39s | |
30m 25s | |
22m 55s | |
7h 19m 28s |