megengine.amp.enabled

enabled = <property object>

Get or set amp autocast mode enabled or not.

实际案例

import megengine as mge
mge.amp.enabled = True