ColossalAI/colossalai/zero/shard_utils
Jiarui Fang 0bebda6ea5
[zero] fix init device bug in zero init context unittest (#516)
2022-03-25 12:24:18 +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 [zero] fix init device bug in zero init context unittest (#516) 2022-03-25 12:24:18 +08:00