ColossalAI/colossalai/zero
Jiarui Fang 0fcfb1e00d
[test] make zero engine test really work (#447)
2022-03-17 17:24:25 +08:00
..
init_ctx use double buffer to handle grad 2022-03-16 14:24:09 +08:00
shard_utils [test] make zero engine test really work (#447) 2022-03-17 17:24:25 +08:00
sharded_model [hotfix] fix initialize bug with zero (#442) 2022-03-17 13:16:22 +08:00
sharded_optim [test] make zero engine test really work (#447) 2022-03-17 17:24:25 +08:00
sharded_param use double buffer to handle grad 2022-03-16 14:24:09 +08:00
__init__.py [zero] hybrid cpu adam (#445) 2022-03-17 15:05:41 +08:00