ColossalAI/model_zoo/moe/models.py at d7ea63992bd2b9dd7de0ba03633d5345b17ac549 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-12 19:06:17 +00:00

Files

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

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

8.4 KiB

Raw Blame History

View Raw