目录
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)