mindspore.amp

梯度缩放

mindspore.amp.LossScaler

使用混合精度时,用于管理损失缩放系数(loss scaler)的抽象类。

mindspore.amp.DynamicLossScaler

动态调整损失缩放系数的管理器。

mindspore.amp.StaticLossScaler

Static Loss scale类。

mindspore.amp.LossScaleManager

使用混合精度时,用于管理损失缩放系数(loss scale)的抽象类。

mindspore.amp.DynamicLossScaleManager

动态调整损失缩放系数的管理器,继承自 mindspore.amp.LossScaleManager

mindspore.amp.FixedLossScaleManager

损失缩放系数不变的管理器,继承自 mindspore.amp.LossScaleManager

数据类型自动转换

mindspore.amp.auto_mixed_precision

对Cell进行自动混合精度处理。

mindspore.amp.build_train_network

构建混合精度训练网络。

溢出检测

mindspore.amp.init_status

初始化溢出状态检测变量。

mindspore.amp.all_finite

检查inputs是否是有效值(无溢出)。