mindspore.amp
梯度缩放
使用混合精度时,用于管理损失缩放系数(loss scaler)的抽象类。 |
|
动态调整损失缩放系数的管理器。 |
|
Static Loss scale类。 |
|
使用混合精度时,用于管理损失缩放系数(loss scale)的抽象类。 |
|
动态调整损失缩放系数的管理器,继承自 |
|
损失缩放系数不变的管理器,继承自 |
数据类型自动转换
对Cell进行自动混合精度处理。 |
|
构建混合精度训练网络。 |
溢出检测
初始化溢出状态检测变量。 |
|
检查inputs是否是有效值(无溢出)。 |