github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-10-21 23:02:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
cb34cd384d7235f3906e8b13a73612955c6413f2
ColossalAI/colossalai/zero/sharded_optim
History
ver217 d0ae0f2215 [zero] update sharded optim v2 (#334)
2022-03-11 15:50:28 +08:00
..
bookkeeping
Feature/zero (#279)
2022-03-11 15:50:28 +08:00
__init__.py
impl shard optim v2 and add unit test
2022-03-11 15:50:28 +08:00
_utils.py
Feature/zero (#279)
2022-03-11 15:50:28 +08:00
sharded_optim_v2.py
[zero] update sharded optim v2 (#334)
2022-03-11 15:50:28 +08:00
sharded_optim.py
[zero] cpu adam kernel (#288)
2022-03-11 15:50:28 +08:00