GPU usage in tune_model.py

Sujit-O / pykg2vec

Python library for knowledge graph embedding and representation learning.

MIT License

606 stars 111 forks source link

GPU usage in tune_model.py #161

Closed vitornl closed 4 years ago

vitornl commented 4 years ago

The -device arg from KGEArgParser class does not affect the KGETuneArgParser class (used in the trune_model.py). Then, how can I enable the use of GPU in the hyperparameter search?

louisccc commented 4 years ago

sorry that I missed that parameter setting. We have modified the code so can you pls try again?

vitornl commented 4 years ago

Thanks, it works just fine!