您好，关于分层计算参数量

[Jerryme-xxm]

您好，请问您用的什么工具分各层计算参数量呢？可以分层计算计算量，如FLOPs吗？

[Berry-Wu]

不好意思，一直没看过这个仓库，我通常使用torchinfo、torchsummary、torchstat这几个包进行计算，具体细节有所差异。其中torchstat可能会满足你的需求