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())
       (20L,)
       (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中怎么查看可训练参数-创新互联
文章路径:http://bzwzjz.com/article/dhcshh.html

其他资讯

Copyright © 2007-2020 广东宝晨空调科技有限公司 All Rights Reserved 粤ICP备2022107769号
友情链接: 重庆网站建设 外贸营销网站建设 成都网站建设公司 成都网站设计公司 定制网站设计 手机网站建设 成都网站设计 企业网站建设公司 成都网站制作 广安网站设计 达州网站设计 成都营销网站制作 成都网站制作 专业网站设计 响应式网站建设 网站设计制作报价 成都模版网站建设 成都网站建设 品牌网站建设 成都网站制作 定制级高端网站建设 古蔺网站建设