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