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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-26 17:53:08 +00:00

Files

History

Jiarui Fang ea2872073f [zero] global model data memory tracer (#360 )

2022-03-11 15:50:28 +08:00

..

__init__.py

[zero] yet an improved sharded param (#311 )

2022-03-11 15:50:28 +08:00

sharded_param.py

[zero] Update sharded model v2 using sharded param v2 (#323 )

2022-03-11 15:50:28 +08:00

sharded_tensor.py

[zero] global model data memory tracer (#360 )

2022-03-11 15:50:28 +08:00