ColossalAI/colossalai/engine
digger yu 09fe9dc704
[nfc]fix ColossalaiOptimizer is not defined (#4122)
2023-06-30 17:23:22 +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
_base_engine.py [nfc]fix ColossalaiOptimizer is not defined (#4122) 2023-06-30 17:23:22 +08:00