分组问题

[yuhbai]

您好，我在使用您的代码对整个mobilenet进行分组时，发现有部分层会被分到不同组里，请问这是为什么

[VainF]

剪枝函数是不一样的。

虽然同一个层出现在两个组，但他们的剪枝方式有差异，比如同一个卷积的输出通道剪枝和输入通道剪枝在依赖图里是两个独立节点。

[yuhbai]

剪枝函数是不一样的。

虽然同一个层出现在两个组，但他们的剪枝方式有差异，比如同一个卷积的输出通道剪枝和输入通道剪枝在依赖图里是两个独立节点。

好的，非常感谢您及时回复。那么如果我想计算每个组的flops应该怎么做呢

[VainF]

可以对整个模型逐层计算好flops，然后对每个组求和。可能需要修改一下现有flops计算工具，存一个dict记录不同层的flops。