ColossalAI/colossalai/zero/sharded_param/__init__.py

4 lines
68 B
Python

from .sharded_param import ShardedParam
__all__ = ['ShardedParam']