Pytorch 获取网络权重参数、每一层权重参数_pytorch打印模型权重

作者：笔触狂放9 | 2024-06-14 07:42:36

踩

pytorch打印模型权重


model = vgg16()
for name in model.state_dict():
  print(name)

 print(model.state_dict()['layers.0.conv2d.weight'])


for name, parameters in model.named_parameters():  
    print(name, ';', parameters.size())


for name, module in model.named_modules():
    print(name)

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/笔触狂放9/article/detail/717073