使用hook提取模型的特征图

目录

一、实例化模型

二、查找想要提取的网络层

1.打印网络层

2.获取网络层的name

三、勾取特征图

1.设置hook函数

2.获取特征图

四、特征图可视化

参考自

---

一、实例化模型

# 通过你的模型实例和输入序列调用函数
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)