mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-10-22 07:14:09 +00:00
5 lines
216 B
Python
5 lines
216 B
Python
from colossalai.zero.shard_utils.base_shard_strategy import BaseShardStrategy
|
|
from colossalai.zero.shard_utils.tensor_shard_strategy import TensorShardStrategy
|
|
|
|
__all__ = ['BaseShardStrategy', 'TensorShardStrategy']
|