ColossalAI/colossalai/engine
digger yu 7f8203af69
fix typo colossalai/auto_parallel autochunk fx/passes etc. (#3808)
2023-05-24 09:01:50 +08:00
..
gradient_accumulation [format] Run lint on colossalai.engine (#3367) 2023-04-05 23:24:43 +08:00
gradient_handler [format] Run lint on colossalai.engine (#3367) 2023-04-05 23:24:43 +08:00
schedule fix typo colossalai/auto_parallel autochunk fx/passes etc. (#3808) 2023-05-24 09:01:50 +08:00
__init__.py fix format (#580) 2022-04-06 11:40:59 +08:00
_base_engine.py [zero] reorganize zero/gemini folder structure (#3424) 2023-04-04 13:48:16 +08:00