Pytorch：summary（）函数打印网络结构

一般我们想知道网络结构是什么样的，有时候看代码会比较麻烦，所以将网络的结构打印出来，会更好的帮助我们理解网络结构，而通常我们会使用print（）函数打印网络结构：

model = AlexNet()
print(model)

运行截图：（以下截图不是AlexNet的网络结构，我随便找的一个网络运行的结果）

但是这样打印出来的网络结构也不太好看。使用summary（）函数进行打印，相对于上述方式打印出来的网络结构会更加简洁：

首先，安装torchinfo库（可以使用pip，也可以在pycharm的setting里下载安装）

from torchinfo import summary

其次，

model = AlexNet()
batch_size = 16
summary(model, input_size=(batch_size, 3, 640, 640))

 截图如下：