ver217 a93a7d7364 [hotfix] fix reuse_fp16_shard of sharded model (#756) ... * fix reuse_fp16_shard * disable test stm * polish code		2022-04-14 14:56:46 +08:00
..
__init__.py	[refactor] zero directory (#724)	2022-04-11 23:13:02 +08:00
sharded_param.py	[hotfix] fix reuse_fp16_shard of sharded model (#756)	2022-04-14 14:56:46 +08:00
sharded_tensor.py	[zero] refactor ShardedParamV2 for convenience (#742)	2022-04-13 14:54:26 +08:00
tensor_utils.py	[zero] refactor memstats_collector (#746)	2022-04-14 12:01:12 +08:00
tensorful_state.py	[zero] refactor ShardedParamV2 for convenience (#742)	2022-04-13 14:54:26 +08:00