ColossalAI/sharded_param at cb34cd384d7235f3906e8b13a73612955c6413f2 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-10-22 07:14:09 +00:00

Files

History

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

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] polish shard strategy (#310 )

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