new TransE framework - Githubissues

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

new TransE framework #176

Closed menjarleev closed 3 years ago

menjarleev commented 3 years ago

Issue #, if available:

num_worker must be set to 0 to prevent initialization of process overwhelm training time.

Description of changes:

new framework with PyTorch implementation.
sampling with Pytorch dataloader

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

dalinvip commented 3 years ago

When is the code merged into the master？

menjarleev commented 3 years ago

When is the code merged into the master？

It should be done in a month. We are refactoring to a more general framework which supports KGE, GNN and so on.

dalinvip commented 3 years ago

When is the code merged into the master？

It should be done in a month. We are refactoring to a more general framework which supports KGE, GNN and so on.

Thanks. It's great. I'm looking forward to it.