ColossalAI/colossalai/engine
Sze-qq 94eec1c5ad [NFC] polish colossalai/engine/gradient_accumulation/_gradient_accumulation.py code style (#3277)
Co-authored-by: siqi <siqi@siqis-MacBook-Pro.local>
2023-03-29 15:22:21 +08:00
..
gradient_accumulation [NFC] polish colossalai/engine/gradient_accumulation/_gradient_accumulation.py code style (#3277) 2023-03-29 15:22:21 +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_v2.py code style (#3275) 2023-03-29 15:22:21 +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