1.打印网络中指定的一层的参数:
print(net.state_dict()['module.conv1.1.bias']) # 填该层参数名
如果不知道要打印的这一层的名称(key),可以通过以下方式获得:
for name in net.state_dict():
print(name)
就会打印出网络每一层的名称
2.打印整个网络每一层的名称和参数值:
for name, parameters in net.named_parameters():
print(name, ':', parameters)
版权声明:本文为qq_45589658原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。