Jiarui Fang 44e4891f57 [zero] able to place params on cpu after zero init context (#365) ... * place params on cpu after zero init context * polish code		2022-03-11 15:50:28 +08:00
..
__init__.py	[zero] a shard strategy in granularity of tensor (#307)	2022-03-11 15:50:28 +08:00
base_shard_strategy.py	[zero] update zero context init with the updated test utils (#327)	2022-03-11 15:50:28 +08:00
tensor_shard_strategy.py	[zero] able to place params on cpu after zero init context (#365)	2022-03-11 15:50:28 +08:00