torch.nn.Module.named_parameters ()

2022-09-02 22:04:38 浏览数 (2)

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

返回模块参数上的迭代器,生成参数的名称和参数本身。

参数:

  • prefix (str) – 在所有参数名称前加上前缀。
  • recurse (bool) – 如果为真,则生成此模块和所有子模块的参数。否则,只生成此模块的直接成员的参数。

生产量:

(string, Parameter) – 包含名称和参数的元组

例:

代码语言:javascript复制
>>> for name, param in self.named_parameters():
>>>    if name in ['bias']:
>>>        print(param.size())

0 人点赞