ColossalAI/colossalai/zero
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