github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-08-31 08:34:14 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
feature/async-io
ColossalAI/colossalai/zero/low_level/bookkeeping
History
Hongxin Liu a2596519fd [zero] support extra dp (#6123) 
* [zero] support extra dp

* [zero] update checkpoint

* fix bugs

* fix bugs
2024-11-12 11:20:46 +08:00
..
__init__.py
[MoE/ZeRO] Moe refactor with zero refactor (#5821)
2024-06-28 14:00:08 +08:00
base_store.py
[zero] support extra dp (#6123)
2024-11-12 11:20:46 +08:00
bucket_store.py
[MoE/ZeRO] Moe refactor with zero refactor (#5821)
2024-06-28 14:00:08 +08:00
gradient_store.py
[FP8] rebase main (#5963)
2024-08-06 16:29:37 +08:00
tensor_bucket.py
[zero] support extra dp (#6123)
2024-11-12 11:20:46 +08:00