ColossalAI

Author	SHA1	Message	Date
HELSON	1a1d68b053	[moe] add checkpoint for moe models (#3354 ) * [moe] add checkpoint for moe models * [hotfix] fix bugs in unit test	2023-03-31 09:20:33 +08:00
HELSON	a088022efc	[moe] fix moe bugs (#1633 )	2022-09-23 15:33:57 +08:00
HELSON	c9023d4078	[MOE] support PR-MOE (#488 )	2022-03-22 16:48:22 +08:00
HELSON	bccbc15861	[MOE] changed parallelmode to dist process group (#460 )	2022-03-19 13:46:29 +08:00
1SAA	82023779bb	Added TPExpert for special situation	2022-03-11 15:50:28 +08:00
1SAA	219df6e685	Optimized MoE layer and fixed some bugs; Decreased moe tests; Added FFNExperts and ViTMoE model	2022-03-11 15:50:28 +08:00
HELSON	dceae85195	Added MoE parallel (#127 )	2022-01-07 15:08:36 +08:00