zjunlp / DeepKE

[EMNLP 2022] An Open Toolkit for Knowledge Graph Extraction and Construction
http://deepke.zjukg.cn/
MIT License
3.52k stars 682 forks source link

RuntimeError: Error(s) in loading state_dict for LM: #449

Closed fuqiang-benz closed 7 months ago

fuqiang-benz commented 7 months ago

你好,我用的是DeepKE-cnSchema开箱即用版 python 3.9 torch 1.13.1 cpu版本 image image 我用的是这里两个模型进行关系抽取然后报错: RuntimeError: Error(s) in loading state_dict for LM: size mismatch for fc.weight: copying a param with shape torch.Size([51, 100]) from checkpoint, the shape in current model is torch.Size([30, 100]). size mismatch for fc.bias: copying a param with shape torch.Size([51]) from checkpoint, the shape in current model is torch.Size([30]). 请问有什么解决办法吗?

还有个疑问 conf/embedding.yaml中num_relations 这个参数意义是什么,是关系的类别数吗?比如说我有关系15类,这个参数就要写15吗,写5会怎么样

zxlzr commented 7 months ago

您好,这是决定抽取多少关系的个数,数字不对可能会导致报错。

fuqiang-benz commented 7 months ago

您好,这是决定抽取多少关系的个数,数字不对可能会导致报错。 好的,我是按照你们官网下载的两个模型,现在问题感觉是模型和预训练模型结构不一致导致的,有什么好的解决办法吗,还是我得bert-base-chinese模型下载了

xxupiano commented 7 months ago

使用训练好的模型进行推理,不需要修改num_relations

xxupiano commented 7 months ago

请问是否将config.yaml中的model改为lm?

fuqiang-benz commented 7 months ago

请问是否将config.yaml中的model改为lm?

已经解决了,感谢感谢