|
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354 |
- mindspore.DynamicLossScaleManager
- ==================================
-
- .. py:class:: mindspore.DynamicLossScaleManager(init_loss_scale=16777216, scale_factor=2, scale_window=2000)
-
- ̬ݶȷŴϵĹ̳ :class:`mindspore.LossScaleManager`
-
- ****
-
- - **init_loss_scale** (float) - ʼݶȷŴϵĬֵ2**24
- - **scale_factor** (int) - Ŵ/СĬֵ2
- - **scale_window** (int) - ʱstepĬֵ2000
-
- ****
-
- >>> from mindspore import Model, nn, DynamicLossScaleManager
- >>>
- >>> net = Net()
- >>> loss_scale_manager = DynamicLossScaleManager()
- >>> optim = nn.Momentum(params=net.trainable_params(), learning_rate=0.1, momentum=0.9)
- >>> model = Model(net, loss_scale_manager=loss_scale_manager, optimizer=optim)
-
- .. py:method:: get_drop_overflow_update()
-
- ֵʾǷڷʱֲ¡
-
- **أ**
-
- boolʼΪTrue
-
- .. py:method:: get_loss_scale()
-
- صǰݶȷŴϵ
-
- **أ**
-
- floatݶȷŴϵ
-
- .. py:method:: get_update_cell()
-
- :class:`mindspore.TrainOneStepWithLossScaleCell` иݶȷŴϵ `Cell` ʵ
-
- **أ**
-
- :class:`mindspore.DynamicLossScaleUpdateCell` ʵڸݶȷŴϵ
-
- .. py:method:: update_loss_scale(overflow)
-
- ״̬ݶȷŴϵСݶȷŴϵݶȷŴϵ
-
- ****
-
- overflow (bool)ʾǷ
-
|