You can not select more than 25 topics
Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
|
- Class mindspore.LossScaleManager
-
- 混合精度梯度放大系数(loss scale)管理器的抽象类。
-
- 派生类需要该类的所有方法。`get_loss_scale`用于获取当前的梯度放大系数。`update_loss_scale`用于更新梯度放大系数,该方法将在训练过程中被调用。`get_update_cell`用于获取更新梯度放大系数的`Cell`实例,该实例在将训练过程中被调用。下沉模式下仅`get_update_cell`方式生效,非下沉模式下两种更新梯度放大系数的方式均生效。
- 例如::class:`mindspore.FixedLossScaleManager`和:class:`mindspore.DynamicLossScaleManager`。
-
-
- get_loss_scale()
- 获取梯度放大系数(loss scale)的值。
-
-
- get_update_cell()
- 获取用于更新梯度放大系数的:class:`mindspore.nn.Cell`实例。
-
- update_loss_scale(overflow)
-
- 根据`overflow`状态更新梯度放大系数(loss scale)。
-
- 参数:
- overflow (bool):表示训练过程是否溢出。
-
|