github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-11 00:56:38 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
4d8a574cd3beb18e497d854e599f2899bc8afa70
ColossalAI/colossalai/zero
History
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_ctx
[zero] add load_state_dict for sharded model (#894)
2022-05-27 10:25:08 +08:00
shard_utils
[gemini] add GeminiMemoryManger (#832)
2022-04-24 13:08:48 +08:00
sharded_model
[zero] add load_state_dict for sharded model (#894)
2022-05-27 10:25:08 +08:00
sharded_optim
[gemini] accelerate adjust_layout() (#878)
2022-04-26 18:08:31 +08:00
sharded_param
[gemini] add GeminiMemoryManger (#832)
2022-04-24 13:08:48 +08:00
utils
[gemini] accelerate adjust_layout() (#878)
2022-04-26 18:08:31 +08:00
__init__.py
[gemini] add GeminiMemoryManger (#832)
2022-04-24 13:08:48 +08:00