ColossalAI/colossalai/zero
ver217 e99af94ab8 rename variables 2022-03-18 14:25:25 +08:00
..
init_ctx update sharded optim and fix zero init ctx 2022-03-18 14:25:25 +08:00
shard_utils [test] make zero engine test really work (#447) 2022-03-17 17:24:25 +08:00
sharded_model update sharded optim and fix zero init ctx 2022-03-18 14:25:25 +08:00
sharded_optim rename variables 2022-03-18 14:25:25 +08:00
sharded_param use double buffer to handle grad 2022-03-16 14:24:09 +08:00
__init__.py update sharded optim and fix zero init ctx 2022-03-18 14:25:25 +08:00