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 [zero] add zero context manager to change config during initialization (#546) 2022-03-29 17:57:59 +08:00
init_context.py [moe] fix MoE bugs (#1628) 2022-09-22 13:56:30 +08:00