mindspore.amp
Loss Scale
Loss scaler abstract class when using mixed precision. |
|
Dynamic Loss scale class. |
|
Static Loss scale class. |
|
Loss scale (Magnification factor of gradients when mix precision is used) manager abstract class when using mixed precision. |
|
Loss scale(Magnification factor of gradients when mix precision is used) manager with loss scale dynamically adjusted, inherits from |
|
Loss scale (Magnification factor of gradients when mix precision is used) manager with a fixed loss scale value, inherits from |
Dtype Autocast
auto mixed precision function. |
|
Build the mixed precision training cell automatically. |
Overflow Detection
Returns a Tensor indicating initialized status for overflow detection. |
|
Returns a scalar Tensor indicating whether the inputs are finite. |