内容
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())
文章出处登录后可见!
已经登录?立即刷新