github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-01 03:45:27 +00:00
Code Issues Packages Projects Releases Wiki Activity
a65cbb7e4e
ColossalAI/colossalai/zero/sharded_param
History
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