Pytorch 打印网络中特定一层的参数

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版权协议,转载请附上原文出处链接和本声明。