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

打印查看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两个方法的差异比较