torch.nn.Module.named_parameters ()

named_parameters(prefix='', recurse=True)[source]

返回模塊參數上的迭代器,生成參數的名稱和參數本身。

參數:

  • prefix (str) – 在所有參數名稱前加上前綴。

  • recurse (bool) – 如果爲真,則生成此模塊和所有子模塊的參數。否則,只生成此模塊的直接成員的參數。

生產量:

(string, Parameter) – 包含名稱和參數的元組

例:

>>> for name, param in self.named_parameters():
>>>    if name in ['bias']:
>>>        print(param.size())

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章