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 |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| checkpoint_io_base.py | ||
| general_checkpoint_io.py | ||
| index_file.py | ||
| utils.py | ||