HELSON 8c90d4df54 [zero] add zero context manager to change config during initialization (#546)

2022-03-29 17:57:59 +08:00

..

init_ctx

[zero] add zero context manager to change config during initialization (#546)

2022-03-29 17:57:59 +08:00

shard_utils

[zero] non model data tracing (#545)

2022-03-29 15:45:48 +08:00

sharded_model

[zero] non model data tracing (#545)

2022-03-29 15:45:48 +08:00

sharded_optim

[zero] non model data tracing (#545)

2022-03-29 15:45:48 +08:00

sharded_param

[zero] get memory usage of sharded optim v2. (#542)

2022-03-29 09:08:18 +08:00

__init__.py

[refactor] remove old zero code (#517)

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