ColossalAI/colossalai/zero/sharded_optim at 38102cf61aece2ce5974dbb09abe2aa298de7f8d - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-12 07:02:22 +00:00

Files

History

ver217 6e553748a7 polish sharded optim docstr and warning (#770 )

2022-04-14 21:03:59 +08:00

..

__init__.py

[refactor] remove old zero code (#517 )

2022-03-25 14:54:39 +08:00

_utils.py

[refactor] zero directory (#724 )

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

sharded_optim_v2.py

polish sharded optim docstr and warning (#770 )

2022-04-14 21:03:59 +08:00