thunlp / OpenNE

An Open-Source Package for Network Embedding (NE)
MIT License
1.68k stars 485 forks source link

Node2vec Library speed #76

Closed ZhiyuanCao98 closed 4 years ago

ZhiyuanCao98 commented 5 years ago

I saw the result of the elapsed time of openNE(node2vec) are slower than the regular library. 1. Could you describe the result of why it will happen?

  1. Can node2vec support multi-gpu

run?

ZhiyuanCao98 commented 5 years ago

I saw the source code, you just directly use the Node2vec library, so it does not support multi-gpu yet

tryggviy commented 5 years ago

I also find that in source codes. The title of this project makes me believe that they have supported node2vec with GPU. They may support it soon.

Bznkxs commented 4 years ago

Hi, Thanks for your discussion! We now support GPU in our new version (https://github.com/thunlp/OpenNE/tree/pytorch) and are currently working on multi-GPU. Please stay tuned!