第一种：查看参数量
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