search

apple55bc / bd-chat-2020

2020语言与智能技术竞赛：面向推荐的对话任务
51 stars 15 forks source link

导入预训练模型问题 #2

Closed jfzhouyoo closed 3 years ago

jfzhouyoo commented 4 years ago

image 您好，在导入了roberta预训练模型的bert_model(包括checkpoint等文件)之后，运行train_bert_lm.py时出现了上面的错误，想请问您一下这可能是什么原因导致的呢？非常感谢您！

apple55bc commented 4 years ago

模型文件格式不对。 去这里下载 https://github.com/ymcui/Chinese-BERT-wwm RoBERTa-wwm-ext, Chinese

jfzhouyoo commented 4 years ago

好的，谢谢您！