按名获取pytorch模型的各参数

想在 pytorch optimizer 对不一样组参数用不一样策略,分组用到参数名字。 获取模型全部参数用 model.parameters(),其中元素是 torch.nn.parameter.Parameter 类型的。 按名访问模型参数有两个途径:python model.state_dict():collections.OrderedDict,拿到的是 torch.Tensor; mod
相关文章
相关标签/搜索