Huggingface预训练的bert模型不起作用

时间:2020-10-15 04:58:18

标签: pytorch bert-language-model huggingface-transformers

我已经用自定义语料库对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文件的目录的正确路径

你能帮我吗?

0 个答案:

没有答案