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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-22 21:24:31 +00:00

Files

History

Jiarui Fang 4d90a7b513 [refactor] zero directory (#724 )

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

..

__init__.py

[refactor] zero directory (#724 )

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

sharded_param.py

[refactor] zero directory (#724 )

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

sharded_tensor.py

[zero] label state for param fp16 and grad (#551 )

2022-03-30 15:57:46 +08:00

tensor_utils.py

[refactor] zero directory (#724 )

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

tensorful_state.py

[zero] label state for param fp16 and grad (#551 )

2022-03-30 15:57:46 +08:00