Neph0s / LMKE

Code for the paper 'Language Models as Knowledge Embeddings'
53 stars 7 forks source link

在train.py中第449行报错如下:这是程序的问题?还是配置环境的问题呢?能麻烦您给出准确的配置环境版本吗 #14

Open star-rui opened 1 year ago

star-rui commented 1 year ago

Traceback (most recent call last): File "/root/LMKE-main/main.py", line 213, in trainer.run() File "/root/LMKE-main/trainer.py", line 96, in run self.train() File "/root/LMKE-main/trainer.py", line 396, in train self.triple_classification(epc) File "/root/LMKE-main/trainer.py", line 449, in triple_classification preds = model(inputs, positions, mode, triple_degrees) File "/opt/conda/envs/pytorch1.8/lib/python3.9/site-packages/torch/nn/modules/module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) TypeError: forward() takes 4 positional arguments but 5 were given

AustynDing commented 1 year ago

Traceback (most recent call last): File "/root/LMKE-main/main.py", line 213, in trainer.run() File "/root/LMKE-main/trainer.py", line 96, in run self.train() File "/root/LMKE-main/trainer.py", line 396, in train self.triple_classification(epc) File "/root/LMKE-main/trainer.py", line 449, in triple_classification preds = model(inputs, positions, mode, triple_degrees) File "/opt/conda/envs/pytorch1.8/lib/python3.9/site-packages/torch/nn/modules/module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) TypeError: forward() takes 4 positional arguments but 5 were given

改成:preds = model(inputs, positions, mode)好像可以正常运行

star-rui commented 1 year ago

是的,谢谢,我把triple_degrees这个参数去掉后,确实可以正常训练了。

twrjie commented 3 months ago

image 我也遇到了类似的问题,请问这样应该怎么改

Neph0s commented 3 months ago

image 我也遇到了类似的问题,请问这样应该怎么改

Sorry,这里有个之前重构代码遗留的bug,我已经修复了,你重新pull试试