ColossalAI/colossalai/zero/utils
ver217 1f894e033f
[gemini] zero supports gemini (#1093)
* add placement policy

* add gemini mgr

* update mem stats collector

* update zero

* update zero optim

* fix bugs

* zero optim monitor os

* polish unit test

* polish unit test

* add assert
2022-06-10 14:48:28 +08:00
..
__init__.py [gemini] init genimi individual directory (#754) 2022-04-14 16:40:26 +08:00
zero_hook.py [gemini] accelerate adjust_layout() (#878) 2022-04-26 18:08:31 +08:00
zero_hook_v2.py [gemini] zero supports gemini (#1093) 2022-06-10 14:48:28 +08:00