mirror of https://github.com/hpcaitech/ColossalAI
152239bbfa
* [gemini] fix nvme optimizer init * [gemini] gemini supports lazy init * [gemini] add init example * [gemini] add fool model * [zero] update gemini ddp * [zero] update init example * add chunk method * add chunk method * [lazyinit] fix lazy tensor tolist * [gemini] fix buffer materialization * [misc] remove useless file * [booster] update gemini plugin * [test] update gemini plugin test * [test] fix gemini plugin test * [gemini] fix import * [gemini] fix import * [lazyinit] use new metatensor * [lazyinit] use new metatensor * [lazyinit] fix __set__ method |
||
---|---|---|
.. | ||
chunk | ||
memory_tracer | ||
__init__.py | ||
colo_init_context.py | ||
gemini_ddp.py | ||
gemini_hook.py | ||
gemini_mgr.py | ||
gemini_optimizer.py | ||
placement_policy.py | ||
utils.py |