search

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.26k stars 194 forks source link

GNN Models #118

Closed snash4 closed 4 years ago

snash4 commented 4 years ago

Hi, Thanks for sharing this library for distributed training. Is there any plan to add GNN models such as RGCN? What needs to be done to add GNN models and train them in a distributed environment.

Thanks

zheng-da commented 4 years ago

Yes, adding GNN models for embeddings is in our roadmap. Currently, we are working on distributed training capabilities in DGL and we are implementing an efficient RGCN in DGL. Once we have these two ready, we'll move the RGCN model in DGL-KE.

snash4 commented 4 years ago

Thanks for the response. What is the possible date of release of DGL with distributed computing capabilities? Further, can I have some more details on the distributed computing side of DGL, such as which distributed framework will be used in DGL? Thanks

zheng-da commented 4 years ago

The distributed training capability will be released with DGL 0.5. The planned release date is the end of July. We didn't put any design documentation yet. DGL builds its own distributed framework.

snash4 commented 4 years ago

Thanks, eagerly waiting for the release. Would love to contribute.