首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >如何从BertForSequenceClassification获取hidden_states?

如何从BertForSequenceClassification获取hidden_states?
EN

Stack Overflow用户
提问于 2020-01-05 00:39:47
回答 1查看 698关注 0票数 2

我阅读了官方教程(https://huggingface.co/transformers/model_doc/bert.html)并尝试设置配置,但它不起作用。

代码语言:javascript
复制
from transformers import PretrainedConfig
model = BertForSequenceClassification.from_pretrained("bert-base-uncased", num_labels=2)
model.config.output_hidden_states = True
model.load_state_dict(torch.load('../parameter.pkl'))
model.cuda()
output = model(input)
EN

回答 1

Stack Overflow用户

发布于 2020-01-05 01:48:59

输出应该是包含隐藏状态的列表。我预计,因为您加载的parameter.pkl在默认情况下可能没有输出隐藏状态,所以它会将您的config.output_hidden_states覆盖为False?加载state_dict后,如果将其设置为True,会发生什么?

票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/59592736

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档