search

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？

已经解决了，感谢感谢