打印查看pytorch 的神经网络模型及参数

最新推荐文章于 2025-01-14 20:04:15 发布

chunfeng0301

最新推荐文章于 2025-01-14 20:04:15 发布

阅读量1.1w

点赞数 10

分类专栏：深度学习文章标签：神经网络深度学习 pytorch

本文链接：https://blog.csdn.net/chunfeng0301/article/details/108227660

版权

深度学习专栏收录该内容

7 篇文章

订阅专栏

本文详细介绍了如何在PyTorch中查看神经网络模型的结构和参数，包括使用parameters()和named_parameters()方法来遍历和打印模型参数，以及利用state_dict()方法获取模型的状态字典。

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

打印查看pytorch 的神经网络模型及参数

net = resnet（）#实例化网络
for parameters in net.parameters():#打印出参数矩阵及值
        print(parameters)

for name, parameters in net.named_parameters():#打印出每一层的参数的大小
       print(name, ':', parameters.size())

for param_tensor in net.state_dict():  # 字典的遍历默认是遍历 key，所以param_tensor实际上是键值
        print(param_tensor, '\t', net.state_dict()[param_tensor].size())

打印查看网络参考这个链接

具体的两个有什么不同如下链接进行了比较state_dict和parameters两个方法的差异比较