2022-08-30 16:52:05 +08:00
|
|
|
mindspore.amp
|
|
|
|
|
================
|
|
|
|
|
|
2022-12-27 15:31:30 +08:00
|
|
|
梯度缩放
|
2022-08-30 16:52:05 +08:00
|
|
|
-----------
|
|
|
|
|
|
|
|
|
|
.. mscnautosummary::
|
|
|
|
|
:toctree: amp
|
|
|
|
|
:nosignatures:
|
|
|
|
|
:template: classtemplate.rst
|
|
|
|
|
|
2022-12-27 15:31:30 +08:00
|
|
|
mindspore.amp.LossScaler
|
|
|
|
|
mindspore.amp.DynamicLossScaler
|
|
|
|
|
mindspore.amp.StaticLossScaler
|
2022-08-30 16:52:05 +08:00
|
|
|
mindspore.amp.LossScaleManager
|
2022-12-27 15:31:30 +08:00
|
|
|
mindspore.amp.DynamicLossScaleManager
|
2022-08-30 16:52:05 +08:00
|
|
|
mindspore.amp.FixedLossScaleManager
|
2022-12-27 15:31:30 +08:00
|
|
|
|
|
|
|
|
数据类型自动转换
|
|
|
|
|
----------------
|
|
|
|
|
|
|
|
|
|
.. mscnautosummary::
|
|
|
|
|
:toctree: amp
|
|
|
|
|
:nosignatures:
|
|
|
|
|
:template: classtemplate.rst
|
|
|
|
|
|
|
|
|
|
mindspore.amp.auto_mixed_precision
|
2022-08-30 16:52:05 +08:00
|
|
|
mindspore.amp.build_train_network
|
|
|
|
|
|
2022-12-27 15:31:30 +08:00
|
|
|
溢出检测
|
2022-08-30 16:52:05 +08:00
|
|
|
-----------
|
|
|
|
|
|
|
|
|
|
.. mscnautosummary::
|
|
|
|
|
:toctree: amp
|
|
|
|
|
:nosignatures:
|
|
|
|
|
:template: classtemplate.rst
|
|
|
|
|
|
|
|
|
|
mindspore.amp.all_finite
|
