pytorch中怎么查看可訓(xùn)練參數(shù)

發(fā)布時(shí)間：2021-06-17 16:59:40 來(lái)源：億速云閱讀：261 作者：Leah 欄目：開(kāi)發(fā)技術(shù)

本篇文章給大家分享的是有關(guān)pytorch中怎么查看可訓(xùn)練參數(shù)，小編覺(jué)得挺實(shí)用的，因此分享給大家學(xué)習(xí)，希望大家閱讀完這篇文章后可以有所收獲，話不多說(shuō)，跟著小編一起來(lái)看看吧。

pytorch中model.parameters()函數(shù)定義如下：

  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()中的所有的參數(shù)，只打印那些param.requires_grad=True的變量。具體實(shí)現(xiàn)代碼如下所示：

for name, param in model.named_parameters():
  if param.requires_grad:
    print(name)

以上就是pytorch中怎么查看可訓(xùn)練參數(shù)，小編相信有部分知識(shí)點(diǎn)可能是我們?nèi)粘９ぷ鲿?huì)見(jiàn)到或用到的。希望你能通過(guò)這篇文章學(xué)到更多知識(shí)。更多詳情敬請(qǐng)關(guān)注億速云行業(yè)資訊頻道。

向AI問(wèn)一下細(xì)節(jié)

pytorch中怎么查看可訓(xùn)練參數(shù)

猜你喜歡

最新資訊

相關(guān)推薦

相關(guān)標(biāo)簽