megengine.amp.high_prec_dtype

high_prec_dtype = <property object>

获取或设置 amp autocast 模式的高精度 dtype。 它将更改 Tensor 转换的目标 dtype 以获得更好的精度。 默认值:float32

实际案例

import megengine as mge
mge.amp.high_prec_dtype = "float32"