mirror of https://github.com/hpcaitech/ColossalAI
![]() * [lazy] remove old lazy init * [lazy] refactor lazy init folder structure * [lazy] fix lazy tensor deepcopy * [test] update lazy init test |
||
---|---|---|
.. | ||
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 |