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.6 kB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253
  1. mindspore.DynamicLossScaleManager
  2. ==================================
  3. .. py:class:: mindspore.DynamicLossScaleManager(init_loss_scale=16777216, 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. ****
  10. >>> from mindspore import Model, nn, DynamicLossScaleManager
  11. >>>
  12. >>> net = Net()
  13. >>> loss_scale_manager = DynamicLossScaleManager()
  14. >>> optim = nn.Momentum(params=net.trainable_params(), learning_rate=0.1, momentum=0.9)
  15. >>> model = Model(net, loss_scale_manager=loss_scale_manager, optimizer=optim)
  16. .. py:method:: get_drop_overflow_update()
  17. ֵʾǷڷʱֲ¡
  18. **أ**
  19. boolʼΪTrue
  20. .. py:method:: get_loss_scale()
  21. صǰݶȷŴϵ
  22. **أ**
  23. floatݶȷŴϵ
  24. .. py:method:: get_update_cell()
  25. ڸݶȷŴϵ `Cell` ʵ:class:`mindspore.TrainOneStepWithLossScaleCell` øʵ
  26. **أ**
  27. :class:`mindspore.DynamicLossScaleUpdateCell` ʵڸݶȷŴϵ
  28. .. py:method:: update_loss_scale(overflow)
  29. ״̬ݶȷŴϵСݶȷŴϵݶȷŴϵ
  30. ****
  31. - **overflow** (bool) - ʾǷ