pytorch获取某层权重_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())