ColossalAI/colossalai/zero/shard_utils
Jiarui Fang 0fcfb1e00d
[test] make zero engine test really work (#447)
2022-03-17 17:24:25 +08:00
..
__init__.py add bucket tensor shard strategy 2022-03-14 14:48:32 +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
bucket_tensor_shard_strategy.py [test] make zero engine test really work (#447) 2022-03-17 17:24:25 +08:00
tensor_shard_strategy.py [test] make zero engine test really work (#447) 2022-03-17 17:24:25 +08:00