Pytorch 可视化模型结构

最新推荐文章于 2023-12-03 18:17:44 发布

原创最新推荐文章于 2023-12-03 18:17:44 发布 · 1.3k 阅读

2 ·

CC 4.0 BY-SA版权

本文详细介绍了ResNet18神经网络模型的结构及参数，通过代码展示了模型的每一层结构及其参数数量，总结了模型的总参数量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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