目录
一、实例化模型
# 通过你的模型实例和输入序列调用函数
pretrained_path = 'path-to-your-pretrained-model'
model = Model() # 替换为你的训练好的模型实例 eg : model = Model("bert", pretrained_path, args, use_cuda=False)
# 加载预训练模型
state = torch.load(pretrained_path, map_location="cpu")
model.load_state_dict(state, strict=False)
二、查找想要提取的网络层
1.打印网络层
查看model的所有属性
#显示model的所有属性的方法
print(dir(model)