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
Returns a network processed with auto mixed precision. |
|
Build the mixed precision training cell automatically. |
|
When the white_list is provided, primitives and cells in white_list will perform the precision conversion. |
|
Provide a copy of internal black list used by auto mixed precision with amp_level set to |
|
Provide a copy of internal white list used by auto mixed precision with amp_level set to |
Overflow Detection
Returns a scalar Tensor indicating whether the inputs are finite. |