ColossalAI/colossalai/engine
ver217 7bf1e98b97
pipeline last stage supports multi output (#151)
2022-01-17 15:57:47 +08:00
..
gradient_handler Added MoE parallel (#127) 2022-01-07 15:08:36 +08:00
schedule pipeline last stage supports multi output (#151) 2022-01-17 15:57:47 +08:00
__init__.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
_base_engine.py Optimize pipeline schedule (#94) 2021-12-30 15:56:46 +08:00