ColossalAI/colossalai/zero/init_ctx
ver217 7cfd6c827e
[zero] add load_state_dict for sharded model (#894)
* add load_state_dict for sharded model

* fix bug

* fix bug

* fix ckpt dtype and device

* support load state dict in zero init ctx

* fix bugs
2022-05-27 10:25:08 +08:00
..
__init__.py [zero] add zero context manager to change config during initialization (#546) 2022-03-29 17:57:59 +08:00
init_context.py [zero] add load_state_dict for sharded model (#894) 2022-05-27 10:25:08 +08:00