Open abasak24 opened 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.
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!