sciai.utils.amp2datatype

sciai.utils.amp2datatype(type_str)

Map from auto mixed precision level string to MindSpore data type. Support amp level from O0 to O3.

Parameters

type_str (str) – Auto mixed precision level string.

Returns

dtype, MindSpore Tensor data type.