ColossalAI/colossalai/zero
ver217 642846d6f9
update sharded optim and fix zero init ctx (#457)
2022-03-18 15:44:47 +08:00
..
init_ctx update sharded optim and fix zero init ctx (#457) 2022-03-18 15:44:47 +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 (#457) 2022-03-18 15:44:47 +08:00
sharded_optim update sharded optim and fix zero init ctx (#457) 2022-03-18 15:44:47 +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 (#457) 2022-03-18 15:44:47 +08:00