search

intfloat / SimKGC

ACL 2022, SimKGC: Simple Contrastive Knowledge Graph Completion with Pre-trained Language Models
180 stars 36 forks source link

Can't run the program #27

Open dqs12345 opened 1 year ago

dqs12345 commented 1 year ago

I encountered the following problem when executing "UTPUT_DIR=./checkpoint/fb15k237/ bash scripts/train_fb.sh".O How can I solve it TypeError: Caught TypeError in replica 0 on device 0. TypeError: forward() got an unexpected keyword argument 'return_dict'

intfloat commented 1 year ago

Have you made any code changes? And what's your version of transformers and pytroch?

I cannot reproduce your issue.

dqs12345 commented 1 year ago

Thank you very much for your reply. I redownloaded the code and configured the environment, and now it can run successfully

lijun-1999 commented 5 months ago

非常感谢您的回复。我重新下载了代码并配置了环境，现在它可以成功运行了

您好，我在运行“OUTPUT_DIR=./checkpoint/wn18rr/ bash scripts/train_wn.sh”命令时，遇到RuntimeError: CUDA out of memory. 的问题，使用的环境是4卡v100，为什么还会遇到这个问题呀？想问您在运行实验时有遇到这个问题吗？是如何解决的呢？