search

mana-ysh / knowledge-graph-embeddings

Implementations of Embedding-based methods for Knowledge Base Completion tasks
Apache License 2.0
257 stars 63 forks source link

GPU implementation #9

Closed JunhengH closed 5 years ago

JunhengH commented 6 years ago

Great work for all the popular KG embedding model!

Are you still working on this project or considering to update another GPU-adaptable version?

mana-ysh commented 5 years ago

Hi @JunhengH Sorry for late reply. I don’t any plan to support GPU. But you may be able to use cupy, which has the same interface as numpy, like this.

import cupy as np

But I didn’t test it.