使用 module_stats（）函数计算flops没有统计上ConvRelu

#1

开发环境：

  • MegEngine 1.3.0
  • Python 3.7
  • Jupyter Notebook

问题详述：
在使用 module_stats（）函数统计模型的参数量和计算量时，flops没有统计上ConvRelu，模型参数量、计算量统计信息和网络结构如下：

~L$(JJY8@V3{)5CYVEUAB
~L$(JJY8@V3{)5CYVEUAB1700×1093 205 KB

718891ca09dfc71c9566c04cdec469d
718891ca09dfc71c9566c04cdec469d861×789 101 KB

#2

请问您用的是什么版本？目前代码里是有统计的，https://github.com/MegEngine/MegEngine/blob/master/imperative/python/megengine/utils/module_stats.py#L41-L64

#3

老师 我用的Megengine1.3.0这个版本
代码如下:

import megengine as mge
from model_MCNN_quantize import MCNN
from megengine.utils import module_stats

mcnn = MCNN()
# print(mcnn)
checkpoint = mge.load('mcnn_net.mge')
mcnn.load_state_dict(checkpoint, strict=False)
# 指定输入shape
input_shape = (1, 3, 1200, 1600)

total_params, total_flops = module_stats.module_stats(
    mcnn, input_shape, log_params=True, log_flops=True
)
print("params: {} flops: {}".format(total_params, total_flops))
#4

感谢您的报告，我们在1.3.1已经解决了该问题，预计本周末发布