DataWhale——《深入浅出Pytorch》3

---

一、可视化网络结构

随着深度学习的发展，许多的网络架构越来越复杂，而且不能很好的理解其中矩阵维度，输出输入维度等信息，对编辑网络架构带来一定的麻烦，因此可以通过torchsummary来对模型的网络架构进行输出，但是该种输出方式存在着一定的缺陷，比如说它只是负责打印出你所搭建的网络有什么有几层，而关于每一层的输入输出并不能完全看得到，因此人们开发了torchinfo工具包 ( torchinfo是由torchsummary和torchsummaryX重构出的库）可以通过以下的指令进行安装

pip install torchinfo

在学习中举了一个关于torchinfo的使用例子

import torchvision.models as models
from torchinfo import summary
resnet18 = models.resnet18() # 实例化模型
summary(model, (1, 3, 224, 224))

代码的结果是

===================================================================