概述

本章节包含各类融合优化器,提供比普通优化器更好的性能。

表1 torch_npu.optim API

API名称

说明

torch_npu.optim.NpuFusedOptimizerBase

通过张量融合实现优化器的基础类,实现梯度清零、梯度更新等优化器基本功能,用户可进行继承实现自定义融合优化器。

torch_npu.optim.NpuFusedSGD

通过张量融合实现的随机梯度下降算法。

torch_npu.optim.NpuFusedAdadelta

通过张量融合实现的 Adadelta 算法。

torch_npu.optim.NpuFusedLamb

通过张量融合实现的 FusedLamb 算法。

torch_npu.optim.NpuFusedAdam

通过张量融合实现的Adam算法。

torch_npu.optim.NpuFusedAdamW

通过张量融合实现的AdamW算法。

torch_npu.optim.NpuFusedAdamP

通过张量融合实现的AdamP算法。

torch_npu.optim.NpuFusedBertAdam

通过张量融合实现的 BertAdam 算法。

torch_npu.optim.NpuFusedRMSprop

通过张量融合实现的 RMSprop 算法。

torch_npu.optim.NpuFusedRMSpropTF

通过张量融合实现的 RMSpropTF 算法。