megengine.amp.convert_module_format

convert_module_format(module, inplace=True)

将 NCHW 模块转换为 NHWC 模块。

上一页

autocast

下一页

megengine.amp.convert_tensor_format