mindspore.amp.LossScaler

class mindspore.amp.LossScaler

使用混合精度时，用于管理损失缩放系数（loss scaler）的抽象类。

派生类需要实现该类的所有方法。训练过程中，scale 和 unscale 用于对损失值或梯度进行放大或缩小，以避免数据溢出；adjust 用于调整损失缩放系数 scale_value 的值。

说明

• 这是一个实验性接口，后续可能删除或修改。

adjust(grads_finite)

根据梯度是否为有效值（无溢出）对 scale_value 进行调整。

参数：

• grads_finite (Tensor) - bool类型的标量Tensor，表示梯度是否为有效值（无溢出）。

scale(inputs)

对inputs进行scale，inputs *= scale_value。

参数：

• inputs (Union(Tensor, tuple(Tensor))) - 损失值或梯度。

unscale(inputs)

对inputs进行unscale，inputs /= scale_value。

参数：

• inputs (Union(Tensor, tuple(Tensor))) - 损失值或梯度。