megengine.amp.convert_module_format

convert_module_format(module, inplace=True)

Convert NCHW Module to NHWC Module.

previous

autocast

next

megengine.amp.convert_tensor_format