ColossalAI/colossalai/zero
ver217 fc8e6db005
[doc] Update docstring for ZeRO (#459)
* polish sharded model docstr

* polish sharded optim docstr

* polish zero docstr

* polish shard strategy docstr
2022-03-18 16:48:20 +08:00
..
init_ctx update sharded optim and fix zero init ctx (#457) 2022-03-18 15:44:47 +08:00
shard_utils [doc] Update docstring for ZeRO (#459) 2022-03-18 16:48:20 +08:00
sharded_model [doc] Update docstring for ZeRO (#459) 2022-03-18 16:48:20 +08:00
sharded_optim [doc] Update docstring for ZeRO (#459) 2022-03-18 16:48:20 +08:00
sharded_param use double buffer to handle grad 2022-03-16 14:24:09 +08:00
__init__.py [zero] Update initialize for ZeRO (#458) 2022-03-18 16:18:31 +08:00