pytorch 实现查看网络中的参数
时间:2020-11-28 14:50:30|栏目:Python代码|点击: 次
可以通过model.state_dict()或者model.named_parameters()函数查看现在的全部可训练参数(包括通过继承得到的父类中的参数)
可示例代码如下:
params = list(model.named_parameters())
(name, param) = params[28]
print(name)
print(param.grad)
print('-------------------------------------------------')
(name2, param2) = params[29]
print(name2)
print(param2.grad)
print('----------------------------------------------------')
(name1, param1) = params[30]
print(name1)
print(param1.grad)
栏 目:Python代码
下一篇:Python linecache.getline()读取文件中特定一行的脚本
本文标题:pytorch 实现查看网络中的参数
本文地址:http://www.codeinn.net/misctech/27322.html






