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.DynamicLossScaleManager.rst 1.2 kB

4 years ago
4 years ago
4 years ago
1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. mindspore.DynamicLossScaleManager
  2. ==================================
  3. .. py:class:: mindspore.DynamicLossScaleManager(init_loss_scale=2 ** 24, scale_factor=2, scale_window=2000)
  4. ̬ʧϵĹ̳ :class:`mindspore.LossScaleManager`
  5. ****
  6. - **init_loss_scale** (float) - ʼݶȷŴϵĬֵ2**24
  7. - **scale_factor** (int) - Ŵ/СĬֵ2
  8. - **scale_window** (int) - ʱstepĬֵ2000
  9. .. py:method:: get_drop_overflow_update()
  10. ֵʾǷڷʱֲ¡
  11. **أ**
  12. boolʼΪTrue
  13. .. py:method:: get_loss_scale()
  14. صǰݶȷŴϵ
  15. **أ**
  16. floatݶȷŴϵ
  17. .. py:method:: get_update_cell()
  18. ڸݶȷŴϵ `Cell` ʵ:class:`mindspore.nn.TrainOneStepWithLossScaleCell` øʵ
  19. **أ**
  20. :class:`mindspore.nn.DynamicLossScaleUpdateCell` ʵڸݶȷŴϵ
  21. .. py:method:: update_loss_scale(overflow)
  22. ״̬ݶȷŴϵСݶȷŴϵݶȷŴϵ
  23. ****
  24. **overflow** (bool) - ʾǷ