ColossalAI/colossalai/zero/init_ctx
HELSON f7f2248771
[moe] fix MoE bugs (#1628)
* remove forced FP32 modules

* correct no_shard-contexts' positions
2022-09-22 13:56:30 +08:00
..
__init__.py
init_context.py [moe] fix MoE bugs (#1628) 2022-09-22 13:56:30 +08:00