ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-02-21 14:32:09 +00:00

Files

Hongxin Liu f313babd11 [gemini] support save state dict in shards (#3581 )

* [gemini] support state dict shard

* [gemini] add test state dict shard

* [gemini] polish docstr

* [gemini] fix merge

* [gemini] polish code

2023-04-17 17:11:09 +08:00

test_chunk_mgrv2.py

2023-04-06 14:51:35 +08:00

test_chunkv2.py

2023-04-06 14:51:35 +08:00

test_fwd_bwd.py

2023-04-06 14:51:35 +08:00

test_gemini_use_rmt.py

2023-04-06 14:51:35 +08:00

test_get_torch_model.py

2023-04-06 14:51:35 +08:00

test_grad_clip.py

2023-04-06 14:51:35 +08:00

test_inference.py

2023-04-06 14:51:35 +08:00

test_optim.py

2023-04-06 14:51:35 +08:00

test_runtime_mem_tracer.py

2023-04-06 14:51:35 +08:00

test_search.py

2023-04-06 14:51:35 +08:00

test_zeroddp_state_dict_shard.py

2023-04-17 17:11:09 +08:00

test_zeroddp_state_dict.py

2023-04-06 14:51:35 +08:00

test_zerooptim_state_dict.py

2023-04-06 14:51:35 +08:00