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://cdiso.cn/article/dhcshh.html