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