github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-03 17:37:23 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
b5b612a26c0291d835778967518a4aa18ffdee85
ColossalAI/model_zoo/moe
History
1SAA e7a1136436 Optimized MoE layer and fixed some bugs; 
Decreased moe tests;

Added FFNExperts and ViTMoE model
2022-02-22 10:27:12 +08:00
..
__init__.py
Added MoE parallel (#127)
2022-01-07 15:08:36 +08:00
models.py
Optimized MoE layer and fixed some bugs;
2022-02-22 10:27:12 +08:00
util.py
Added MoE parallel (#127)
2022-01-07 15:08:36 +08:00