search

DreamInvoker / GAIN

Source code for EMNLP 2020 paper: Double Graph Based Reasoning for Document-level Relation Extraction
MIT License
142 stars 31 forks source link

Training cannot be done in GPU #13

Closed otpas007 closed 3 years ago

otpas007 commented 3 years ago

Hi! I'm trying to train the neural network, using the default values provided in the script run_GAIN_GLOVE.sh. The code is running fine but I found that the training is being done using CPU. I tried to change the devices in the code to cuda to make it train using GPU. But it didn't work and started throwing errors "GPU device not available" although my system has GPUs.

Can you please help to execute the code using GPU?

DreamInvoker commented 3 years ago

We set gpu_id in the shell script, please run the code following the intructions in the README.md.