按名獲取pytorch模型的各參數

想在 pytorch optimizer 對不一樣組參數用不一樣策略,分組用到參數名字。 獲取模型全部參數用 model.parameters(),其中元素是 torch.nn.parameter.Parameter 類型的。 按名訪問模型參數有兩個途徑:python model.state_dict():collections.OrderedDict,拿到的是 torch.Tensor; mod
相關文章
相關標籤/搜索