mindspore.amp.LossScaler
- class mindspore.amp.LossScaler[源代码]
使用混合精度时,用于管理损失缩放系数(loss scaler)的抽象类。
派生类需要实现该类的所有方法。训练过程中,scale 和 unscale 用于对损失值或梯度进行放大或缩小,以避免数据溢出;adjust 用于调整损失缩放系数 scale_value 的值。
Note
这是一个实验性接口,后续可能删除或修改。
- adjust(grads_finite)[源代码]
根据梯度是否为有效值(无溢出)对 scale_value 进行调整。
- 参数:
grads_finite (Tensor) - bool类型的标量Tensor,表示梯度是否为有效值(无溢出)。