ColossalAI/colossalai/zero
ver217 0ef8819c67
polish docstring of zero (#612)
2022-04-01 14:50:56 +08:00
..
init_ctx [zero] adapt zero for unsharded parameters (#561) 2022-03-31 18:34:11 +08:00
shard_utils [refactor] memory utils (#577) 2022-04-01 09:22:33 +08:00
sharded_model polish docstring of zero (#612) 2022-04-01 14:50:56 +08:00
sharded_optim polish docstring of zero (#612) 2022-04-01 14:50:56 +08:00
sharded_param [refactor] memory utils (#577) 2022-04-01 09:22:33 +08:00
__init__.py