mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-10-21 23:02:07 +00:00
b334822163
* [zero] polish sharded param name * polish code * polish * polish code * polish * polsih * polish
5 lines
189 B
Python
5 lines
189 B
Python
from colossalai.zero.sharded_param.sharded_tensor import ShardedTensor
|
|
from colossalai.zero.sharded_param.sharded_param import ShardedParamV2
|
|
|
|
__all__ = ['ShardedTensor', 'ShardedParamV2']
|