Closed Bznkxs closed 3 years ago
Thank you for your work! But I still don't find how can we run this packages on gpu, where should we adjust the script?
Thank you for your work! But I still don't find how can we run this packages on gpu, where should we adjust the script?
@LynnZhang9 please refer to readme.md at the pytorch branch(https://github.com/thunlp/OpenNE/tree/pytorch). It should be straightforward to run the stable branch with GPU.
Supports GPU; Modifies model structure: