torch_npu.optim
- 概述
- torch_npu.optim.NpuFusedOptimizerBase
- torch_npu.optim.NpuFusedSGD
- torch_npu.optim.NpuFusedAdadelta
- torch_npu.optim.NpuFusedLamb
- torch_npu.optim.NpuFusedAdam
- torch_npu.optim.NpuFusedAdamW
- torch_npu.optim.NpuFusedAdamP
- torch_npu.optim.NpuFusedBertAdam
- torch_npu.optim.NpuFusedRMSprop
- torch_npu.optim.NpuFusedRMSpropTF