wangle1218 / KBQA-for-Diagnosis

Knowledge Graph,Question Answering System,基于知识图谱和向量检索的医疗诊断问答系统
1.33k stars 380 forks source link

运行时报错 #10

Open Okiii-lh opened 3 years ago

Okiii-lh commented 3 years ago

在代码中如下面路径的文件,请问这些文件需要到哪里下载呀,这个没有找到呢

self.dict_path = 'E:/bert_weight_files/roberta/vocab.txt' self.config_path='E:/bert_weight_files/roberta/bert_config_rbt3.json' self.checkpoint_path='E:/bert_weight_files/roberta/bert_model.ckpt'

wangle1218 commented 3 years ago

这是一个三层的roberta预训练模型,官方有开源,github应该有,你搜一下

Okiii-lh commented 3 years ago

收到,谢谢

Okiii-lh commented 3 years ago

ValueError: You are trying to load a weight file containing 21 layers into a model with 57 layers.

wangle1218 commented 3 years ago

ValueError: You are trying to load a weight file containing 21 layers into a model with 57 layers.

模型不对吗。。。嗯。。我晚上回去看一下我的是多少层吧。。

wangle1218 commented 3 years ago

http://pan.iflytek.com/#/link/275E5B46185C982D4AF5AC295E1651B6 试试这个

Okiii-lh commented 3 years ago

ValueError: You are trying to load a weight file containing 21 layers into a model with 57 layers.

模型不对吗。。。嗯。。我晚上回去看一下我的是多少层吧。。

是我找的模型不对,已经解决了,谢谢啦,跑app.py时内存爆了,不知道您的硬件设备是什么样子的

wangle1218 commented 3 years ago

ValueError: You are trying to load a weight file containing 21 layers into a model with 57 layers.

模型不对吗。。。嗯。。我晚上回去看一下我的是多少层吧。。

是我找的模型不对,已经解决了,谢谢啦,跑app.py时内存爆了,不知道您的硬件设备是什么样子的

我是32G内存,别人8G的内存也能跑啊,占内存的就一个bert模型吧。。

Stajo3 commented 2 years ago

http://pan.iflytek.com/#/link/275E5B46185C982D4AF5AC295E1651B6 试试这个 请问这个讯飞网盘提取密码是多少? 我自己下的下载RBT3,Chinese 版本,有三个,不知道是哪一个bert_model.ckpt.meta,bert_model.ckpt.index,bert_model.ckpt.data-00000-of-00001