ColossalAI/colossalai/zero/sharded_param/__init__.py

from colossalai.zero.sharded_param.sharded_tensor import ShardedTensor
from colossalai.zero.sharded_param.sharded_param import ShardedParamV2

__all__ = [
    'ShardedTensor', 'ShardedParamV2']
[zero] sharded tensor (#305) * 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 2022-03-04 02:46:13 +00:00			`from colossalai.zero.sharded_param.sharded_tensor import ShardedTensor`
[zero] polish sharded param name (#484) * [zero] polish sharded param name * polish code * polish * polish code * polish * polsih * polish 2022-03-22 06:36:16 +00:00			`from colossalai.zero.sharded_param.sharded_param import ShardedParamV2`
Polish sharded parameter (#297) * init shard param from shape tuple * add more unitest for shard param * add more unittests to shareded param 2022-03-03 04:42:57 +00:00
[refactor] zero directory (#724) 2022-04-11 15:13:02 +00:00			`__all__ = [`
[gemini] add GeminiMemoryManger (#832) * refactor StatefulTensor, tensor utilities * add unitest for GeminiMemoryManager 2022-04-24 05:08:48 +00:00			`'ShardedTensor', 'ShardedParamV2']`