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

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354
  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. :class:`mindspore.TrainOneStepWithLossScaleCell` иݶȷŴϵ `Cell` ʵ
  26. **أ**
  27. :class:`mindspore.DynamicLossScaleUpdateCell` ʵڸݶȷŴϵ
  28. .. py:method:: update_loss_scale(overflow)
  29. ״̬ݶȷŴϵСݶȷŴϵݶȷŴϵ
  30. ****
  31. overflow (bool)ʾǷ