Can't run the program

[dqs12345]

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]

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

I cannot reproduce your issue.

[dqs12345]

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

[lijun-1999]

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

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