Queries related to the library for KG_Completion/Text Classification

graph4ai / graph4nlp

Graph4nlp is the library for the easy use of Graph Neural Networks for NLP. Welcome to visit our DLG4NLP website (https://dlg4nlp.github.io/index.html) for various learning resources!

Apache License 2.0

1.67k stars 201 forks source link

Queries related to the library for KG_Completion/Text Classification #431

Closed sandeepKondaveeti closed 3 years ago

sandeepKondaveeti commented 3 years ago

❓ Questions and Help

Can we use Numerical Labels for Text classification ?
Is GPU Mandatory to pre-process KG Completion?

Feel free to ask questions.

Nicoleqwerty commented 3 years ago

Sorry, we forgot to test KG Completion on the CPU. If you want to run KGC on a CPU, you can make the following changes in the kinship_main.py:

change 'cuda' to 'cpu' in line 167 https://github.com/graph4ai/graph4nlp/blob/master/examples/pytorch/kg_completion/kinship_main.py#L167
comment line 181
chang 'Config.cuda = False' to 'Config.cuda = True' in line 370

Thank you for your advice. We will fix it in next version.

AlanSwift commented 3 years ago

I will close this issue since the bug has been fixed. Please feel free to reopen if you still have problems.