【DL】torch小技巧之网络参数统计 torchstat & torchsummary

1 torchstat 优缺点

优点：直接就可以使用，打印所以结构，信息详细，名字详细。
缺点：按照类的顺序进行打印，循环中未使用的结果也打印了。

1.1 安装工具包torchstat

pip install torchstat
1

1.2 测试参数

from torchstat import stat

# 导入模型，输入一张输入图片的尺寸
stat(model, (3, 224, 224))
1
2
3
4

1.3 结果

CCMBlk(
  (relu): ReLU(inplace=True)
  (conv1): Conv2d(3, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
  (conv2): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
  (maxpool): MaxPool2d(kernel_size=3, stride=2, padding=1, dilation=1, ceil_mode=False)
)
      module name  input shape output shape   params memory(MB)          MAdd         Flops  MemRead(B)  MemWrite
1
2
3
4
5
6