在pytorch中查看可训练参数的例子
更新时间：2019年08月18日 10:56:50 作者：玩人
今天小编就为大家分享一篇在pytorch中查看可训练参数的例子，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
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中查看可训练参数的例子就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
今天小编就为大家分享一篇python每5分钟从kafka中提取数据的例子，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-12-12
这篇文章主要介绍了在Django中进行用户注册和邮箱验证的方法的相关资料,需要的朋友可以参考下2016-05-05
这篇文章主要介绍了python3 打印输出字典中特定的某个key的方法,涉及Python字典的遍历、判断、输出等相关操作技巧,需要的朋友可以参考下2019-07-07
饼图,或称饼状图,是一个划分为几个扇形的圆形统计图表,这篇文章主要为大家详细介绍了如何使用Matplotlib绘制饼图,需要的小伙伴可以参考下2023-08-08
这篇文章主要介绍了python 如何在list中找Topk的数值和索引的操作，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2021-05-05
这篇文章主要介绍了Python closure闭包解释及其注意点详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-08-08
观察者模式：是一种行为型设计模式。主要关注的是对象的责任，允许你定义一种订阅机制，可在对象事件发生时通知多个"观察"该对象的其他对象。本文将利用观察者模式实现日志实时监测，需要的可以参考一下2022-04-04
这篇文章主要介绍了pandas实现按照多列排序-ascending，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2022-05-05
缓存是一种用于提高应用程序性能的技术,它通过临时存储程序获得的结果,以便在以后需要时重用它们,本文将学习Python中的不同缓存技术,感兴趣的可以了解下2024-11-11
这篇文章主要介绍了20个实用的Python Excel自动化脚本,用于自动化处理和分析Excel文件,提高工作效率,从批量填充单元格、设置行高与列宽到生成数据透视表和图表,涵盖了数据处理和分析的各个方面,需要的朋友可以参考下2025-02-02
最新评论