github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/engine
History
HELSON aff9d354f7
[MOE] polish moe_env (#467) 		2022-03-19 15:36:25 +08:00
..
gradient_handler [MOE] polish moe_env (#467) 2022-03-19 15:36:25 +08:00
ophooks [zero] Update initialize for ZeRO (#458) 2022-03-18 16:18:31 +08:00
paramhooks Fix/format colossalai/engine/paramhooks/(#350) 2022-03-11 15:50:28 +08:00
schedule Feature/zero (#279) 2022-03-11 15:50:28 +08:00
__init__.py
…
_base_engine.py set criterion as optional in colossalai initialize (#336) 2022-03-11 15:50:28 +08:00