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 |
||
|---|---|---|
| .. | ||
| checkpoint | ||
| checkpoint_io | ||
| data_sampler | ||
| model | ||
| multi_tensor_apply | ||
| profiler | ||
| rank_recorder | ||
| tensor_detector | ||
| __init__.py | ||
| activation_checkpoint.py | ||
| checkpointing.py | ||
| common.py | ||
| cuda.py | ||
| memory.py | ||
| moe.py | ||
| timer.py | ||