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

最新推荐文章于 2024-03-29 17:55:31 发布

张林克

最新推荐文章于 2024-03-29 17:55:31 发布

阅读量2w

点赞数 14

分类专栏：【DeepL】

本文链接：https://blog.csdn.net/weixin_45292794/article/details/108227437

版权

1 torchstat 优缺点

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

1.1 安装工具包torchstat

pip install torchstat

1.2 测试参数

from torchstat import stat

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

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