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.

mindspore.LossScaleManager.txt 995 B

12345678910111213141516171819202122
  1. Class mindspore.LossScaleManager
  2. 混合精度梯度放大系数(loss scale)管理器的抽象类。
  3. 派生类需要该类的所有方法。`get_loss_scale`用于获取当前的梯度放大系数。`update_loss_scale`用于更新梯度放大系数,该方法将在训练过程中被调用。`get_update_cell`用于获取更新梯度放大系数的`Cell`实例,该实例在将训练过程中被调用。下沉模式下仅`get_update_cell`方式生效,非下沉模式下两种更新梯度放大系数的方式均生效。
  4. 例如::class:`mindspore.FixedLossScaleManager`和:class:`mindspore.DynamicLossScaleManager`。
  5. get_loss_scale()
  6. 获取梯度放大系数(loss scale)的值。
  7. get_update_cell()
  8. 获取用于更新梯度放大系数的:class:`mindspore.nn.Cell`实例。
  9. update_loss_scale(overflow)
  10. 根据`overflow`状态更新梯度放大系数(loss scale)。
  11. 参数:
  12. overflow (bool):表示训练过程是否溢出。