github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-01 20:33:02 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
7e30068a22bb3bcc09bb92014b77e7e9dfa4573d
ColossalAI/colossalai/zero/sharded_model
History
ver217 fc8e6db005 [doc] Update docstring for ZeRO (#459) 
* polish sharded model docstr

* polish sharded optim docstr

* polish zero docstr

* polish shard strategy docstr
2022-03-18 16:48:20 +08:00
..
__init__.py
Feature/zero (#279)
2022-03-11 15:50:28 +08:00
_zero3_utils.py
[zero] Update sharded model v2 using sharded param v2 (#323)
2022-03-11 15:50:28 +08:00
param_manager.py
Feature/zero (#279)
2022-03-11 15:50:28 +08:00
reduce_scatter.py
[zero] add sharded grad and refactor grad hooks for ShardedModel (#287)
2022-03-11 15:50:28 +08:00
sharded_grad.py
[zero] add sharded grad and refactor grad hooks for ShardedModel (#287)
2022-03-11 15:50:28 +08:00
sharded_model_v2.py
[doc] Update docstring for ZeRO (#459)
2022-03-18 16:48:20 +08:00
sharded_model.py
[zero] Update sharded model v2 using sharded param v2 (#323)
2022-03-11 15:50:28 +08:00
utils.py
[test] polish zero related unitest (#351)
2022-03-11 15:50:28 +08:00