欢迎来到代码驿站!

Python代码

当前位置:首页 > 软件编程 > 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)

上一篇:Python同时迭代多个序列的方法

栏    目:Python代码

下一篇:Python linecache.getline()读取文件中特定一行的脚本

本文标题:pytorch 实现查看网络中的参数

本文地址:http://www.codeinn.net/misctech/27322.html

推荐教程

广告投放 | 联系我们 | 版权申明

重要申明:本站所有的文章、图片、评论等,均由网友发表或上传并维护或收集自网络,属个人行为,与本站立场无关。

如果侵犯了您的权利,请与我们联系,我们将在24小时内进行处理、任何非本站因素导致的法律后果,本站均不负任何责任。

联系QQ:914707363 | 邮箱:codeinn#126.com(#换成@)

Copyright © 2020 代码驿站 版权所有