mindspore.amp
Cell Management
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 abstract class when using mixed precision. |
|
Loss scale (Magnification factor of gradients when mix precision is used) manager with a fixed loss scale value, inherits from |
|
Build the mixed precision training cell automatically. |
Functional Paradigm
Dynamic Loss scale class. |
|
Static Loss scale class. |
|
Loss scaler abstract class when using mixed precision. |
|
auto mixed precision function. |
|
Returns a Tensor indicating initialized status for overflow detection. |
|
Returns a scalar Tensor indicating whether the inputs are finite. |