megengine.amp.enabled

enabled = <property object>

获取或设置是否启用 amp autocast.

实际案例

import megengine as mge
mge.amp.enabled = True