1) 可视化模型结构
from torchviz import make_dot
model = ResNet18()
print(model) #输出模型的详细信息
x = torch.randn(1,3,32, 32).requires_grad_(True)
y = model(x)
vis_graph = make_dot(y, params=dict(list(model.named_parameters()) + [('x', x)]))
vis_graph.view()
exit()执行完上述代码，将可视化后的图像，以pdf格式保存到当前.py文件的路径下。
2）将定义好的模型详细信息列出来
Code：
net = ResNet18()
#print(ResNet18())
x = torch.randn(1, 3, 32, 32).requires_grad_(True)
y = net(x)params = list(net.parameters())
k = 0
for i in params:
l = 1
print("该层的结构：" + str(list(i.size())))
for j in i.size():
l *= j
print("该层参数和：" + str(l))
k = k + l
print("总参数数量和：" + str(k))
exit()Result：
该层的结构：[64, 3, 3, 3]
该层参数和：1728
该层的结构：[64]
该层参数和：64
该层的结构：[64]
该层参数和：64
该层的结构：[64, 64, 3, 3]
该层参数和：36864
该层的结构：[64]
该层参数和：64
该层的结构：[64]
该层参数和：64
该层的结构：[64, 64, 3, 3]
该层参数和：36864
该层的结构：[64]
该层参数和：64
该层的结构：[64]
该层参数和：64
该层的结构：[64, 64, 3, 3]
该层参数和：36864
该层的结构：[64]
该层参数和：64
该层的结构：[64]
该层参数和：64
该层的结构：[64, 64, 3, 3]
该层参数和：36864
该层的结构：[64]
该层参数和：64
该层的结构：[64]
该层参数和：64
该层的结构：[128, 64, 3, 3]
该层参数和：73728
该层的结构：[128]
该层参数和：128
该层的结构：[128]
该层参数和：128
该层的结构：[128, 128, 3, 3]
该层参数和：147456
该层的结构：[128]
该层参数和：128
该层的结构：[128]
该层参数和：128
该层的结构：[128, 64, 1, 1]
该层参数和：8192
该层的结构：[128]
该层参数和：128
该层的结构：[128]
该层参数和：128
该层的结构：[128, 128, 3, 3]
该层参数和：147456
该层的结构：[128]
该层参数和：128
该层的结构：[128]
该层参数和：128
该层的结构：[128, 128, 3, 3]
该层参数和：147456
该层的结构：[128]
该层参数和：128
该层的结构：[128]
该层参数和：128
该层的结构：[256, 128, 3, 3]
该层参数和：294912
该层的结构：[256]
该层参数和：256
该层的结构：[256]
该层参数和：256
该层的结构：[256, 256, 3, 3]
该层参数和：589824
该层的结构：[256]
该层参数和：256
该层的结构：[256]
该层参数和：256
该层的结构：[256, 128, 1, 1]
该层参数和：32768
该层的结构：[256]
该层参数和：256
该层的结构：[256]
该层参数和：256
该层的结构：[256, 256, 3, 3]
该层参数和：589824
该层的结构：[256]
该层参数和：256
该层的结构：[256]
该层参数和：256
该层的结构：[256, 256, 3, 3]
该层参数和：589824
该层的结构：[256]
该层参数和：256
该层的结构：[256]
该层参数和：256
该层的结构：[512, 256, 3, 3]
该层参数和：1179648
该层的结构：[512]
该层参数和：512
该层的结构：[512]
该层参数和：512
该层的结构：[512, 512, 3, 3]
该层参数和：2359296
该层的结构：[512]
该层参数和：512
该层的结构：[512]
该层参数和：512
该层的结构：[512, 256, 1, 1]
该层参数和：131072
该层的结构：[512]
该层参数和：512
该层的结构：[512]
该层参数和：512
该层的结构：[512, 512, 3, 3]
该层参数和：2359296
该层的结构：[512]
该层参数和：512
该层的结构：[512]
该层参数和：512
该层的结构：[512, 512, 3, 3]
该层参数和：2359296
该层的结构：[512]
该层参数和：512
该层的结构：[512]
该层参数和：512
该层的结构：[10, 512]
该层参数和：5120
该层的结构：[10]
该层参数和：10
总参数数量和：11173962
