github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-26 01:35:21 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
e7a1136436db5324dc7925485de625cc3cbb3969
ColossalAI/tests/test_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
..
short_test.py
Optimized MoE layer and fixed some bugs;
2022-02-22 10:27:12 +08:00
test_top1.py
Optimized MoE layer and fixed some bugs;
2022-02-22 10:27:12 +08:00
test_top2.py
Optimized MoE layer and fixed some bugs;
2022-02-22 10:27:12 +08:00