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

好的,谢谢您!