knowledge graph embedding examples - Bin Shang

hpcaitech / ColossalAI-Examples

Examples of training models with hybrid parallelism using ColossalAI

Apache License 2.0

334 stars 102 forks source link

knowledge graph embedding examples - Bin Shang #40

Closed MiracleDesigner closed 2 years ago

MiracleDesigner commented 2 years ago

Add three knowledge graph embedding examples DistMult, ComplEx and RotatE.

MiracleDesigner commented 2 years ago

@FrankLeeeee please check the code.

MiracleDesigner commented 2 years ago

@FrankLeeeee Could you deal with this PR as soon as possible? Because this is a task assigned to me by Professor Yong You, I need to reply to him as soon as possible. Thank you very much.