42 lines
835 B
ReStructuredText
42 lines
835 B
ReStructuredText
mindspore.amp
|
|
================
|
|
|
|
Loss Scale
|
|
----------------
|
|
|
|
.. autosummary::
|
|
:toctree: amp
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
mindspore.amp.LossScaler
|
|
mindspore.amp.DynamicLossScaler
|
|
mindspore.amp.StaticLossScaler
|
|
mindspore.amp.LossScaleManager
|
|
mindspore.amp.DynamicLossScaleManager
|
|
mindspore.amp.FixedLossScaleManager
|
|
|
|
Dtype Autocast
|
|
--------------------
|
|
|
|
.. autosummary::
|
|
: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
|
|
|
|
Overflow Detection
|
|
--------------------
|
|
|
|
.. autosummary::
|
|
:toctree: amp
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
mindspore.amp.all_finite
|