Pytorch 获取网络权重参数、每一层权重参数

目录

1、查看每层对应的名称

2、输出相应层的权重

3、打印模块名字和参数大小

1、查看每层对应的名称

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

2、输出相应层的权重

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

3、打印模块名字和参数大小

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

4、打印模块结构和模块名

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