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