ColossalAI/colossalai/zero/sharded_param at 3ddbd1bce127dedf982caf8129f36d43f9ecf054 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-25 01:03:35 +00:00

Files

History

HELSON 4c4388c46e [hotfix] fix memory leak in zero (#781 )

2022-04-18 13:57:03 +08:00

..

__init__.py

[refactor] zero directory (#724 )

2022-04-11 23:13:02 +08:00

sharded_param.py

[hotfix] fix memory leak in zero (#781 )

2022-04-18 13:57:03 +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