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.FixedLossScaleManager.rst 1.5 kB

4 years ago
4 years ago
12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. mindspore.FixedLossScaleManager
  2. ===============================
  3. .. py:class:: mindspore.FixedLossScaleManager(loss_scale=128.0, drop_overflow_update=True)
  4. ʧϵĹ̳ :class:`mindspore.LossScaleManager`
  5. ****
  6. - **loss_scale** (float) - ݶȷŴϵע `drop_overflow_update` ΪFalseŻʱҪŻ `loss_scale` ΪֵͬĬֵ128.0
  7. - **drop_overflow_update** (bool) - ʱǷִŻֵΪTrueʱִŻĬֵTrue
  8. .. py:method:: get_drop_overflow_update()
  9. `drop_overflow_update` ֵʾǷڷʱֲ¡
  10. **أ**
  11. bool, `drop_overflow_update` ֵ
  12. .. py:method:: get_loss_scale()
  13. ȡloss scaleֵ
  14. **أ**
  15. bool`loss_scale` ֵ
  16. .. py:method:: get_update_cell()
  17. ڸ `loss_scale` ֵ `Cell` ʵ :class:`mindspore.nn.TrainOneStepWithLossScaleCell` øʵʹù̶ݶȷŴϵ˸ʵִκβ
  18. **أ**
  19. None `Cell` `drop_overflow_update` ΪTrueʱ :class:`mindspore.nn.FixedLossScaleUpdateCell` ʵ `drop_overflow_update` ΪFalseʱNone
  20. .. py:method:: update_loss_scale(overflow)
  21. loss scaleֵ :class:`mindspore.FixedLossScaleManager` У÷ִκβ
  22. ****
  23. - **overflow** (bool) - ʾǷ