torch.cuda.amp
序号 |
API名称 |
npu对应API名称 |
是否支持 |
|---|---|---|---|
1 |
torch.cuda.amp.GradScaler |
torch.npu.amp.GradScaler |
否 |
父主题： PyTorch1.5.0
序号 |
API名称 |
npu对应API名称 |
是否支持 |
|---|---|---|---|
1 |
torch.cuda.amp.GradScaler |
torch.npu.amp.GradScaler |
否 |