ColossalAI/colossalai/zero/init_ctx
YH 80aed29cd3
[zero] Refactor ZeroContextConfig class using dataclass (#3186)
2023-03-21 12:36:47 +08:00
..
__init__.py [zero] add zero context manager to change config during initialization (#546) 2022-03-29 17:57:59 +08:00
init_context.py [zero] Refactor ZeroContextConfig class using dataclass (#3186) 2023-03-21 12:36:47 +08:00