LuGY 1ff7d5bfa5 [NFC] polish colossalai/engine/gradient_handler/_moe_gradient_handler.py (#3260)

2023-03-29 15:22:21 +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/_moe_gradient_handler.py (#3260)

2023-03-29 15:22:21 +08:00

schedule

[NFC] polish colossalai/engine/schedule/_pipeline_schedule.py code style (#2744)

2023-02-20 10:38:40 +08:00

__init__.py

fix format (#580)

2022-04-06 11:40:59 +08:00

_base_engine.py

[doc] add deepspeed citation and copyright (#2996)

2023-03-04 20:08:11 +08:00