mirror of https://github.com/hpcaitech/ColossalAI
df63564184
* [gemini] support no reuse fp16 chunk * [gemini] support no master weight for optim * [gemini] support no master weight for gemini ddp * [test] update gemini tests * [test] update gemini tests * [plugin] update gemini plugin * [test] fix gemini checkpointio test * [test] fix gemini checkpoint io |
||
---|---|---|
.. | ||
__init__.py | ||
comparison.py | ||
pytest_wrapper.py | ||
random.py | ||
utils.py |