zjunlp/Relphormer

Entity Embedding Initialization

mqEpiphany opened this issue · 6 comments

mqEpiphany commented

Hello, I found the following error while performing entity embedding initialization, how can I change the code?
image

bizhen46766 commented

Hello, you can try to run the python file with single GPU

mqEpiphany commented

I only have one gpu

bizhen46766 commented

Is your CUDA available in the current python interpreter setting?
you can check by the code, torch.cuda.is_available()

bizhen46766 commented

Hi, I think this problem might be caused by the inaccurate use of the cuda device.
You can first check the pytorch_lightning package version and then make sure the correct use of cuda id.

mqEpiphany commented

Thanks for your help, I have solved this problem by modifying the code of transformers/tokenization utils base.py