mirror of https://github.com/hpcaitech/ColossalAI
58913441a1
* [checkpointio] unsharded optimizer checkpoint for Gemini plugin * [checkpointio] unsharded optimizer checkpoint for Gemini using all_gather |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| comparison.py | ||
| pytest_wrapper.py | ||
| random.py | ||
| utils.py | ||