昇腾社区首页
中文
注册
开发者
下载

torch_npu.optim.NpuFusedOptimizerBase

产品支持情况

[object Object]undefined

功能说明

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

函数原型

[object Object]

参数说明

  • params ([object Object]):必选参数,模型参数或模型参数组。
  • default ([object Object]):包含其他所有参数的字典。

返回值说明

类型为[object Object]的对象。

约束说明

[object Object]为基类,无法单独使用,需通过继承子类实现特定功能的融合优化器。

调用示例

[object Object]