mirror of https://github.com/hpcaitech/ColossalAI
c6f6005990
* sharded optimizer checkpoint for gemini plugin * modify test to reduce testing time * update doc * fix bug when keep_gatherd is true under GeminiPlugin |
||
|---|---|---|
| .. | ||
| test_gemini_checkpoint_io.py | ||
| test_gemini_torch_compability.py | ||
| test_general_checkpoint_io.py | ||
| test_low_level_zero_checkpoint_io.py | ||
| test_torch_ddp_checkpoint_io.py | ||
| test_torch_fsdp_checkpoint_io.py | ||
| utils.py | ||