使用方法

  1. 模型从GPU适配到NPU时，需要将代码torch.cuda.amp修改为torch_npu.npu.amp。
  2. 当前PyTorch1.8.1 AMP工具中GradScaler增加了dynamic选项（默认为True），设置为False时，AMP能支持静态Loss Scale。
父主题： AMP使用说明