当前位置:主页 > 软件编程 > Python代码 >

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)

您可能感兴趣的文章:

相关文章