ColossalAI/colossalai/zero/shard_utils
ver217 1388671699 [zero] Update sharded model v2 using sharded param v2 (#323) 2022-03-11 15:50:28 +08:00
..
__init__.py [zero] a shard strategy in granularity of tensor (#307) 2022-03-11 15:50:28 +08:00
base_shard_strategy.py [zero] update zero context init with the updated test utils (#327) 2022-03-11 15:50:28 +08:00
tensor_shard_strategy.py [zero] Update sharded model v2 using sharded param v2 (#323) 2022-03-11 15:50:28 +08:00