mindspore2022/docs/api/api_python/mindspore/mindspore.FixedLossScaleManager.rst

mindspore.FixedLossScaleManager
===============================

.. py:class:: mindspore.FixedLossScaleManager(loss_scale=128.0, drop_overflow_update=True)

    �ݶȷŴ�ϵ������Ĺ��������̳��� :class:`mindspore.LossScaleManager` ��

    **������**

    - **loss_scale** (float) - �ݶȷŴ�ϵ����ע������� `drop_overflow_update` ��ΪFalse�������Ż���ʱ��Ҫ���Ż����� `loss_scale` ��Ϊ��ͬ��ֵ��Ĭ��ֵ��128.0��
    - **drop_overflow_update** (bool) - �������ʱ���Ƿ�ִ���Ż��������ֵΪTrue����������ʱ����ִ���Ż�����Ĭ��ֵ��True��

    **������**

    >>> from mindspore import Model, nn, FixedLossScaleManager
    >>>
    >>> net = Net()
    >>> # 1) ��������������ִ�в�������
    >>> loss_scale_manager = FixedLossScaleManager()
    >>> 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)
    >>>
    >>> # 2) ��ʹ���������Ҳִ�в�������
    >>> loss_scale = 1024.0
    >>> loss_scale_manager = FixedLossScaleManager(loss_scale, False)
    >>> optim = nn.Momentum(params=net.trainable_params(), learning_rate=0.1, momentum=0.9, loss_scale=loss_scale)
    >>> model = Model(net, loss_scale_manager=loss_scale_manager, optimizer=optim)

    .. py:method:: get_drop_overflow_update()

        ���� `drop_overflow_update` ����ֵ��ʾ�Ƿ��ڷ������ʱ�������ֲ������¡�

        **���أ�**

        bool, `drop_overflow_update` ��ֵ��

    .. py:method:: get_loss_scale()

        ��ȡloss scaleֵ��

        **���أ�**

        bool��`loss_scale` ��ֵ��

    .. py:method:: get_update_cell()

        �������ڸ��� `loss_scale` ֵ�� `Cell` ʵ���� :class:`mindspore.TrainOneStepWithLossScaleCell` ����ø�ʵ��������ʹ�ù̶����ݶȷŴ�ϵ������˸�ʵ����ִ���κβ�����

        **���أ�**

        None�� `Cell` ���� `drop_overflow_update` ΪTrueʱ������ :class:`mindspore.FixedLossScaleUpdateCell` ʵ������ `drop_overflow_update` ΪFalseʱ������None��

    .. py:method:: update_loss_scale(overflow)

        ����loss scaleֵ���� :class:`mindspore.FixedLossScaleManager` �У��÷�����ִ���κβ�����

        **������**

        - **overflow** (bool) - ��ʾ�Ƿ������