mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-04-11 06:33:30 +00:00
e17e92c54d
* init shard param from shape tuple * add more unitest for shard param * add more unittests to shareded param
4 lines
68 B
Python
4 lines
68 B
Python
from .sharded_param import ShardedParam
|
|
|
|
__all__ = ['ShardedParam']
|