ColossalAI/colossalai/zero/init_ctx
Jiarui Fang 67b4928244
[zero] polish init context (#645)
2022-04-02 15:52:04 +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] polish init context (#645) 2022-04-02 15:52:04 +08:00