TransE model utilizing Colossal-AI

hpcaitech / ColossalAI-Examples

Examples of training models with hybrid parallelism using ColossalAI

Apache License 2.0

334 stars 102 forks source link

Closed CsRic closed 2 years ago

CsRic commented 2 years ago

This example uses Colossal-AI engine to predict entities of a Knowledge Graph. It can run on a laptop with one cuda core for now.