mindspore.amp
梯度缩放
| 使用混合精度时,用于管理损失缩放系数(loss scaler)的抽象类。 | |
| 用于动态调整损失缩放系数的管理器。 | |
| Static Loss scale类。 | |
| 使用混合精度时,用于管理损失缩放系数(loss scale)的抽象类。 | |
| 动态调整损失缩放系数的管理器,继承自  | |
| 损失缩放系数不变的管理器,继承自  | 
数据类型自动转换
| 返回一个经过自动混合精度处理的网络。 | |
| 构建混合精度训练网络。 | |
| 通过配置白名单或黑名单,对Cell进行自定义混合精度处理。 | |
| 提供用于自动混合精度 amp_level 为  | |
| 提供用于自动混合精度 amp_level 为  | 
溢出检测
| 检查输入是否有溢出,返回为True为溢出,False为无溢出。 |