ColossalAI/colossalai/zero/legacy/sharded_model
digger yu de0d7df33f
[nfc] fix typo colossalai/zero (#3923)
2023-06-08 00:01:29 +08:00
..
__init__.py [zero] reorganize zero/gemini folder structure (#3424) 2023-04-04 13:48:16 +08:00
_utils.py [bf16] add bf16 support (#3882) 2023-06-05 15:58:31 +08:00
reduce_scatter.py [zero] reorganize zero/gemini folder structure (#3424) 2023-04-04 13:48:16 +08:00
sharded_model_v2.py [nfc] fix typo colossalai/zero (#3923) 2023-06-08 00:01:29 +08:00
utils.py [zero] reorganize zero/gemini folder structure (#3424) 2023-04-04 13:48:16 +08:00
zero_hook.py [zero] reorganize zero/gemini folder structure (#3424) 2023-04-04 13:48:16 +08:00