mindspore.amp

Loss Scale

mindspore.amp.LossScaler

Loss scaler abstract class when using mixed precision.

mindspore.amp.DynamicLossScaler

Dynamic Loss scale class.

mindspore.amp.StaticLossScaler

Static Loss scale class.

mindspore.amp.LossScaleManager

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

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.FixedLossScaleManager

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

Dtype Autocast

mindspore.amp.auto_mixed_precision

auto mixed precision function.

mindspore.amp.build_train_network

Build the mixed precision training cell automatically.

Overflow Detection

mindspore.amp.init_status

Returns a Tensor indicating initialized status for overflow detection.

mindspore.amp.all_finite

Returns a scalar Tensor indicating whether the inputs are finite.