查看pytorch模型种权重的名字及值

调用模型的named_modules()方法

import torch
model_path = ''
model = torch.load(model_path )
named_modules = model.named_modules()## 获取所有模块的名称
name_params = model.named_parameters()## 获取有权值的模块的名称及其权值
params = model.parameters()## 只有权值
print(dir(model))
for k, v in model.named_modules():
    print(k)

输出：

查看模型权值model.parameters()

for k in model.parameters():
    print(k)

输出：