docs/api/api_python/mindspore.amp.rst - master - envoy.istio/mindspore

docs/api/api_python/mindspore.amp.rst

41 строка835 B

huodagu

modify format

25 мар 2023, 04:40

25 мар 2023, 04:4007fece1

mindspore.amp
================

梯度缩放
-----------

.. mscnautosummary::
    :toctree: amp
    :nosignatures:
    :template: classtemplate.rst

    mindspore.amp.LossScaler
    mindspore.amp.DynamicLossScaler
    mindspore.amp.StaticLossScaler
    mindspore.amp.LossScaleManager
    mindspore.amp.DynamicLossScaleManager
    mindspore.amp.FixedLossScaleManager

数据类型自动转换
----------------

.. mscnautosummary::
    :toctree: amp
    :nosignatures:
    :template: classtemplate.rst

    mindspore.amp.auto_mixed_precision
    mindspore.amp.build_train_network
    mindspore.amp.custom_mixed_precision
    mindspore.amp.get_black_list
    mindspore.amp.get_white_list

溢出检测
-----------

.. mscnautosummary::
    :toctree: amp
    :nosignatures:
    :template: classtemplate.rst

    mindspore.amp.all_finite