module_dump

功能说明

开启模块级精度数据dump。

模块级精度数据dump时必选。

原型

module_dump(module, module_name)
表1 参数说明

参数名

说明

是否必选

module

网络中实例化好的nn.Module类模块的model对象。

module_name

用户自定义的该model名称。主要用于dump数据文件的命名,便于在比对时识别模块级数据。