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

Fixed edge_weight dimension #248

Closed moudheus closed 2 years ago

moudheus commented 2 years ago

Fix proposal for Issue #242

Resized the weight vector to match pytorch broadcasting rule.

classicsong commented 2 years ago

LGTM