megengine.amp¶
A helper class that performs grad scaling to prevent from data overflow in |
|
A class to control autocast mode for amp as a context manager or a decorator. |
|
Convert NCHW Module to NHWC Module. |
|
Convert NCHW Tensor to NHWC Tensor. |
获取或设置是否启用 amp autocast. |
|
Get or set amp autocast mode's higher precision dtype. |
|
Get or set amp autocast mode's lower precision dtype. |