ColossalAI/colossalai/zero/utils
HELSON 7829aa094e
[ddp] add is_ddp_ignored (#2434)
[ddp] rename to is_ddp_ignored
2023-01-11 12:22:45 +08:00
..
__init__.py [gemini] init genimi individual directory (#754) 2022-04-14 16:40:26 +08:00
gemini_hook.py [ddp] add is_ddp_ignored (#2434) 2023-01-11 12:22:45 +08:00
zero_hook.py [Gemini] update the non model data record method in runtime memory tracer (#2128) 2022-12-13 17:11:31 +08:00