文件最后提交记录最后更新时间
26 天前
1 个月前
18 天前
26 天前
16 小时前
1 个月前
18 天前
README

LambUpdateWithLrV2

产品支持情况

产品 是否支持
Ascend 950PR/Ascend 950DT
Atlas A3 训练系列产品/Atlas A3 推理系列产品 ×
Atlas A2 训练系列产品/Atlas A2 推理系列产品 ×
Atlas 200I/500 A2 推理产品 ×
Atlas 推理系列产品 ×
Atlas 训练系列产品 ×
Kirin X90 处理器系列产品 ×
Kirin 9030 处理器系列产品 ×

功能说明

  • 算子功能:BERT LAMB优化器图融合算子(信任比权重更新,无裁剪):以阈值greater_y与回退值select_e计算信任比,再用学习率x3更新参数x5。

  • 计算公式:

    ratio=where(x1>greater_y, where(x2>greater_y, x1/x2, select_e), select_e)ratio = where(x1>greater\_y,\ where(x2>greater\_y,\ x1/x2,\ select\_e),\ select\_e)

    y=x5−x3×ratio×x4y = x5 - x3 \times ratio \times x4

参数说明

参数名 输入/输出 描述 数据类型 数据格式
x1 输入 不支持空Tensor。公式中的x1(权重范数),标量。 FLOAT16、FLOAT ND
x2 输入 不支持空Tensor。公式中的x2(梯度范数),标量。 FLOAT16、FLOAT ND
x3 输入 不支持空Tensor。公式中的x3(学习率),标量。 FLOAT16、FLOAT ND
x4 输入 支持空Tensor。公式中的x4(update),主张量,shape需与x5满足broadcast关系。 FLOAT16、FLOAT ND
x5 输入 支持空Tensor。公式中的x5(参数),主张量,shape需与x4满足broadcast关系,其broadcast结果决定输出y的shape。 FLOAT16、FLOAT ND
greater_y 输入 不支持空Tensor。公式中的greater_y(阈值),标量。 FLOAT16、FLOAT ND
select_e 输入 不支持空Tensor。公式中的select_e(回退值),标量。 FLOAT16、FLOAT ND
y 输出 支持空Tensor。公式中的y(更新后的参数),shape取x4与x5的broadcast结果。 FLOAT16、FLOAT ND

约束说明

  • 所有输入的数据类型必须一致,同为FLOAT16或同为FLOAT。

调用说明

调用方式 调用样例 说明
图模式调用 test_geir_lamb_update_with_lr_v2 通过算子IR构图方式调用LambUpdateWithLrV2算子。