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
父主题:
PyTorch2.1