torch_npu.optim.NpuFusedOptimizerBase
产品支持情况
功能说明
张量融合优化器的基类,实现梯度清零、梯度更新等优化器基本功能,用户可进行继承实现自定义融合优化器。
函数原型
[object Object]
参数说明
- params (
[object Object]):必选参数,模型参数或模型参数组。 - default (
[object Object]):包含其他所有参数的字典。
返回值说明
类型为[object Object]的对象。
约束说明
[object Object]为基类,无法单独使用,需通过继承子类实现特定功能的融合优化器。
调用示例
[object Object]