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

返回一个经过自动混合精度处理的网络。

mindspore.amp.build_train_network

构建混合精度训练网络。

mindspore.amp.custom_mixed_precision

通过配置白名单或黑名单,对Cell进行自定义混合精度处理。

mindspore.amp.get_black_list

提供用于自动混合精度 amp_levelO2 等级的内置黑名单的拷贝。

mindspore.amp.get_white_list

提供用于自动混合精度 amp_levelO1 等级的内置白名单的拷贝。

溢出检测

mindspore.amp.all_finite

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