在pytorch中查看可训练参数的例子
作者:玩人 时间:2021-09-18 09:44:06
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)
这样打印出的结果就是模型中所有的可训练参数列表!
来源:https://blog.csdn.net/jeryjeryjery/article/details/83057199
标签:pytorch,查看,训练,参数
0
投稿
猜你喜欢
python中range()与xrange()用法分析
2021-03-23 00:31:30
Python读取配置文件的实战操作
2021-08-12 19:48:09
聊聊python中令人迷惑的duplicated和drop_duplicates()用法
2022-01-03 19:10:57
python opencv根据颜色进行目标检测的方法示例
2021-09-29 03:53:41
Python利用format函数实现对齐打印(左对齐、右对齐与居中对齐)
2021-07-30 05:16:16
PHP convert_uudecode()函数讲解
2023-06-12 15:53:51
SQL提高查询效率之Like篇
2011-10-01 09:36:42
Python学习pygal绘制线图代码分享
2022-12-02 10:51:48
Python在Matplotlib图中显示中文字体的操作方法
2023-03-30 00:29:54
运行SQL Server的计算机间移动数据库
2009-01-20 13:07:00
怎样用cmd命令行运行Python文件
2023-07-15 00:25:11
sql 存储过程分页代码 支持亿万庞大数据量
2011-09-30 11:16:46
Python基于HOG+SVM/RF/DT等模型实现目标人行检测功能
2021-07-12 01:54:50
如何自动备份Oracle数据库
2008-06-13 16:54:00
Python socket 套接字实现通信详解
2023-05-16 04:18:08
python子类在多继承中使用MRO机制原理
2021-03-09 00:52:35
把vgg-face.mat权重迁移到pytorch模型示例
2021-11-03 16:29:20
asp如何读取一个文件内容?
2009-11-19 17:23:00
Django web框架使用url path name详解
2022-01-06 19:43:59
Python OpenCV视频截取并保存实现代码
2023-01-05 06:04:21