Open star-rui opened 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)好像可以正常运行
是的,谢谢,我把triple_degrees这个参数去掉后,确实可以正常训练了。
我也遇到了类似的问题,请问这样应该怎么改
我也遇到了类似的问题,请问这样应该怎么改
Sorry,这里有个之前重构代码遗留的bug,我已经修复了,你重新pull试试
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