feat_act = {}
def get_activation(name):
def hook(model,input,output):
feat_act[name] = output.detach()
return hook
model(训练好的模型).avgpool(层的名字).rgister_forward_hook[用来抓取的函数,参数是get_activation](get_activation('模型名字'))
钩子函数输出某一层hook
最新推荐文章于 2023-03-29 09:04:12 发布