github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/zero/shard_utils
History
HELSON 88759e289e
[zero] add ZeroTensorShardStrategy (#793) 		2022-04-19 14:32:45 +08:00
..
__init__.py [zero] add ZeroTensorShardStrategy (#793) 2022-04-19 14:32:45 +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 [zero] refactor model data tracing (#537) 2022-03-28 16:38:18 +08:00
commons.py [zero] refactor shard and gather operation (#773) 2022-04-15 14:41:31 +08:00
tensor_shard_strategy.py [zero] refactor shard and gather operation (#773) 2022-04-15 14:41:31 +08:00
zero_tensor_shard_strategy.py [zero] add ZeroTensorShardStrategy (#793) 2022-04-19 14:32:45 +08:00