Examples of Web information processing

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

TransE and KGCN. tensor parallelism can be applied.

yuxuan-lou commented 2 years ago

Hi @CsRic , thank you for your contribution! Can you provide some train logs of these two examples?

yuxuan-lou commented 2 years ago

I noticed that you use torch dataloader for training in KGCN example. Have you tried colossalai.utils.get_dataloader?

binmakeswell commented 2 years ago

This PR is closed due to inactivity.