如何解决Huggingface预训练的bert模型不起作用
我已经用自定义语料库对bert模型进行了预训练,然后得到了vocab文件,检查点,model.bin,tfrecords等。
然后我按如下方式加载模型:
# Load pre-trained model (weights)
model = BertModel.from_pretrained('/content/drive/My Drive/Anirban_test_pytorch')
但是,当我尝试将模型用于任何任务(例如q和a,掩码字的预测等)时,错误下方的错误提示
from transformers import pipeline
nlp = pipeline("fill-mask",model="model")
nlp(f"This is the best thing I've {nlp.tokenizer.mask_token} in my life.")
错误:
OSError:无法加载“模型”的配置。确保:
-
“模型”是在'https://huggingface.co/models'
上列出的正确模型标识符 -
或'model'是包含config.json文件的目录的正确路径
你能帮我吗?
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。