mindspore.LossScaleManager
- class mindspore.LossScaleManager[源代码]
使用混合精度时,用于管理损失缩放系数(loss scale)的抽象类。
派生类需要实现该类的所有方法。 get_loss_scale 用于获取当前的梯度放大系数。 update_loss_scale 用于更新梯度放大系数,该方法将在训练过程中被调用。 get_update_cell 用于获取更新梯度放大系数的 Cell 实例,该实例将在训练过程中被调用。当前多使用 get_update_cell 方式。
例如:
mindspore.FixedLossScaleManager
和mindspore.DynamicLossScaleManager
。