可视化网络模型结构

第一种：查看参数量

pip install torchsummary

#此段代码之显示Output Shape和Param

from torchsummary import summary

summary(model.cuda(), input_size=(3, 224, 224))

224分别为输入高度和宽度，3为通道数，界面如下所示

第2种：查看参数量，这种的最详细，但是也是最新的方法，所以还在开发开发当中

from torchstat import stat

stat(model, input_size=(3, 224, 224))

如果出现报错，请看我另一篇文章写的https://blog.csdn.net/qq_44666320/article/details/106094205

，可以解决这个报错的原因，其他的报错就不知道了