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

Reduce some memory overhead #224

Closed classicsong closed 3 years ago

classicsong commented 3 years ago

220

cdpierse commented 3 years ago

@classicsong This looks great, thank you for working on this, it will be a huge help for me :-)