mindspore.amp

Cell Management

mindspore.amp.DynamicLossScaleManager

Loss scale(Magnification factor of gradients when mix precision is used) manager with loss scale dynamically adjusted, inherits from mindspore.amp.LossScaleManager.

mindspore.amp.LossScaleManager

Loss scale (Magnification factor of gradients when mix precision is used) manager abstract class when using mixed precision.

mindspore.amp.FixedLossScaleManager

Loss scale(Magnification factor of gradients when mix precision is used) manager with a fixed loss scale value, inherits from mindspore.amp.LossScaleManager.

mindspore.amp.build_train_network

Build the mixed precision training cell automatically.

Functional Paradigm

mindspore.amp.DynamicLossScaler

Dynamic Loss scale class.

mindspore.amp.StaticLossScaler

Static Loss scale class.

mindspore.amp.LossScaler

Loss scaler abstract class when using mixed precision.

mindspore.amp.auto_mixed_precision

auto mixed precision function.

mindspore.amp.all_finite

Returns a scalar Tensor indicating whether the inputs are finite.