Use torch.amp instead of torch.cuda.amp (#877)
#177
| Job | Run time |
|---|---|
| 5m 59s | |
| 5m 59s |
torch.amp instead of torch.cuda.amp (#877)
#177
| Job | Run time |
|---|---|
| 5m 59s | |
| 5m 59s |