mindspore.amp
Loss Scale
Loss scaler abstract class when using mixed precision. |
|
Manager for dynamically adjusting the loss scaling factor. |
|
Static Loss scale class. |
|
Loss scale (magnification factor of gradients when mixed precision is used) manager abstract class when using mixed precision. |
|
Loss scale (magnification factor of gradients when mixed precision is used) manager with loss scale dynamically adjusted, inherits from |
|
Loss scale (magnification factor of gradients when mixed 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. |
|
Customize mixed precision processing for Cells by configuring a whitelist or a blacklist. |
|
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. |