mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-10-22 07:14:09 +00:00
80364c7686
* init shard param from shape tuple * add more unitest for shard param * add set_payload method for ShardedParam * [zero] add shareded tensor class * polish code
5 lines
185 B
Python
5 lines
185 B
Python
from colossalai.zero.sharded_param.sharded_param import ShardedParam
|
|
from colossalai.zero.sharded_param.sharded_tensor import ShardedTensor
|
|
|
|
__all__ = ['ShardedParam', 'ShardedTensor']
|