CZYCW 4ac8bfb072 [NFC] polish colossalai/engine/gradient_handler/utils.py code style (#2708)		2023-02-15 09:40:08 +08:00
..
gradient_accumulation	[engine] fixed bug in gradient accumulation dataloader to keep the last step (#1030)	2022-05-26 14:28:23 +08:00
gradient_handler	[NFC] polish colossalai/engine/gradient_handler/utils.py code style (#2708)	2023-02-15 09:40:08 +08:00
schedule	[engin/schedule] use p2p_v2 to recontruct pipeline_schedule (#1408)	2022-08-12 11:33:26 +08:00
__init__.py	fix format (#580)	2022-04-06 11:40:59 +08:00
_base_engine.py	[hotfix] remove potiential circle import (#1307)	2022-07-14 13:44:26 +08:00