在pytorch中查看可训练参数的例子
2019-08-18 10:56
1031 查看
pytorch中我们有时候可能需要设定某些变量是参与训练的,这时候就需要查看哪些是可训练参数,以确定这些设置是成功的。
pytorch中model.parameters()函数定义如下:
def parameters(self): r"""Returns an iterator over module parameters. This is typically passed to an optimizer. Yields: Parameter: module parameter Example:: >>> for param in model.parameters(): >>> print(type(param.data), param.size()) <class 'torch.FloatTensor'> (20L,) <class 'torch.FloatTensor'> (20L, 1L, 5L, 5L) """ for name, param in self.named_parameters(): yield param
所以,我们可以遍历named_parameters()中的所有的参数,只打印那些param.requires_grad=True的变量。具体实现代码如下所示:
for name, param in model.named_parameters(): if param.requires_grad: print(name)
这样打印出的结果就是模型中所有的可训练参数列表!
以上这篇在pytorch中查看可训练参数的例子就是小编分享给大家的全部内容了,希望能给大家一个参考
您可能感兴趣的文章:
相关文章推荐
- pytorch中查看可训练参数
- python查看CNN训练模型参数
- matlab训练好神经网络之后,查看其权值参数。
- Pytorch加载部分预训练模型的参数实例
- 深度学习框架里使用GPU训练网络参数时查看GPU的使用情况
- 解决了PyTorch 使用torch.nn.DataParallel 进行多GPU训练的一个BUG:模型(参数)和数据不在相同设备上
- pytorch 在网络中添加可训练参数,修改预训练权重文件的方法
- C#中调用SQL存储过程(带输入输出参数的例子)
- window.open() 支持环境,语法,参数,例子(详细)
- Linux如何查看系统当前登录用户信息,linux su命令参数及用法详解(linux切换用户命令)
- 如何针对已经安装好的Apache/PHP/Mysql/Nginx程序查看他们的编译参数
- 查看nginx,apache,mysql,php的编译参数
- 数组作为参数的一个例子
- Linux RPM 命令参数使用详解 查看 rpm包依赖性
- 通过查看mysql 配置参数、状态来优化你的mysql
- linux中netstat查看80端口连接数例子
- Tensorflow CIFAR-10训练例子报错解决
- 在XP下是可以查看进程命令行参数的。使用下面的命令。
- nginx服务器中url重写rewrite参数和例子
- crontab 各参数详解及如何查看日志记录