在pytorch中查看可训练参数的例子
时间：2020-11-29 10:13:13|栏目：Python代码|点击： 次
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)
这样打印出的结果就是模型中所有的可训练参数列表！
上一篇：python re正则表达式模块(Regular Expression)
栏 目：Python代码
本文标题：在pytorch中查看可训练参数的例子
本文地址：http://www.codeinn.net/misctech/27337.html
