|
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253 |
- 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()
-
- ڸݶȷŴϵ `Cell` ʵ:class:`mindspore.TrainOneStepWithLossScaleCell` øʵ
-
- **أ**
-
- :class:`mindspore.DynamicLossScaleUpdateCell` ʵڸݶȷŴϵ
-
- .. py:method:: update_loss_scale(overflow)
-
- ״̬ݶȷŴϵСݶȷŴϵݶȷŴϵ
-
- ****
-
- - **overflow** (bool) - ʾǷ
|