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