for name in net.state_dict():
print(name)
print(net.state_dict()[name]) #. 参数
for name, parameters in net.named_parameters():
print(name, ':', parameters)
for (name, module) in net.named_modules():
print(name, ':', module)
for name, layer in net._modules.items():
Pytorch 打印网络中某层的参数
最新推荐文章于 2023-04-10 16:09:53 发布