torch.cuda.amp

表1 torch.cuda.amp

序号

API名称

npu对应API名称

是否支持

1

torch.cuda.amp.GradScaler

torch.npu.amp.GradScaler

否

父主题： PyTorch1.5.0