github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-15 14:47:16 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
3cb3fc275e39152a7bcc60eb6b65dfc614e91456
ColossalAI/colossalai/zero/shard_utils
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
add bucket tensor shard strategy
2022-03-14 14:48:32 +08:00
base_shard_strategy.py
[zero] Update initialize for ZeRO (#458)
2022-03-18 16:18:31 +08:00
bucket_tensor_shard_strategy.py
[doc] Update docstring for ZeRO (#459)
2022-03-18 16:48:20 +08:00
tensor_shard_strategy.py
[doc] Update docstring for ZeRO (#459)
2022-03-18 16:48:20 +08:00