torch_npu.npu.amp
Torch_npu提供amp相关的部分接口,具体可参考
torch.cuda.amp
。
父主题:
PyTorch2.1