Closed JunhengH closed 5 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?
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.
Great work for all the popular KG embedding model!
Are you still working on this project or considering to update another GPU-adaptable version?