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

目录

1、查看每层对应的名称

2、输出相应层的权重

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


1、查看每层对应的名称

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

Pytorch 获取网络权重参数、每一层权重参数_第1张图片

2、输出相应层的权重

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

Pytorch 获取网络权重参数、每一层权重参数_第2张图片

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

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

Pytorch 获取网络权重参数、每一层权重参数_第3张图片

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

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

Pytorch 获取网络权重参数、每一层权重参数_第4张图片

你可能感兴趣的:(技巧分享,pytorch,深度学习,python)