github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/tests/test_moe
History
HELSON 84fd7c1d4d
add moe context, moe utilities and refactor gradient handler (#455) 		2022-03-18 16:38:32 +08:00
..
short_test.py add moe context, moe utilities and refactor gradient handler (#455) 2022-03-18 16:38:32 +08:00
test_top1.py Optimized MoE layer and fixed some bugs; 2022-03-11 15:50:28 +08:00
test_top2.py Optimized MoE layer and fixed some bugs; 2022-03-11 15:50:28 +08:00