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))
代码的结果是
====================================================================