ColossalAI/model_zoo/moe at 7f5e4592eb0e8918b596debf09af53cecc446059 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-12 10:54:49 +00:00

Files

History

HELSON 7544347145 [MOE] add unitest for MOE experts layout, gradient handler and kernel (#469 )

2022-03-21 13:35:04 +08:00

..

__init__.py

Added MoE parallel (#127 )

2022-01-07 15:08:36 +08:00

models.py

[MOE] add unitest for MOE experts layout, gradient handler and kernel (#469 )

2022-03-21 13:35:04 +08:00

util.py

Added MoE parallel (#127 )

2022-01-07 15:08:36 +08:00