awslabs / dgl-ke

High performance, easy-to-use, and scalable package for learning large-scale knowledge graph embeddings.
https://dglke.dgl.ai/doc/
Apache License 2.0
1.28k stars 196 forks source link

Reference RGCN implementation for multi-gpu training #137

Open abasak24 opened 4 years ago

abasak24 commented 4 years ago

Hi, Is there any reference implementation of RGCN on knowledge graphs for multi-gpu training in DGL-KE? Alternatively, in DGL, I can see some reference RGCN implementation for link prediction task, but I think that is not multi-gpu (may I confirm if that is correct?). Do you have some suggestions on how I can run RGCN on a multi-gpu setting? Thanks for some pointers on this issue!

zheng-da commented 4 years ago

We don't have RGCN implementation in DGL-KE right now. We have one multi-GPU version in DGL: https://github.com/dmlc/dgl/blob/master/examples/pytorch/rgcn/entity_classify_mp.py. We'll have a distributed implementation very soon.