ColossalAI/colossalai/zero/legacy/gemini
Hongxin Liu ac178ca5c1 [legacy] move builder and registry to legacy (#4603) 2023-09-05 21:53:10 +08:00
..
ophooks [legacy] move builder and registry to legacy (#4603) 2023-09-05 21:53:10 +08:00
paramhooks [zero] reorganize zero/gemini folder structure (#3424) 2023-04-04 13:48:16 +08:00
__init__.py [zero] reorganize zero/gemini folder structure (#3424) 2023-04-04 13:48:16 +08:00
gemini_context.py [zero] reorganize zero/gemini folder structure (#3424) 2023-04-04 13:48:16 +08:00
stateful_tensor.py [zero] reorganize zero/gemini folder structure (#3424) 2023-04-04 13:48:16 +08:00
stateful_tensor_mgr.py [doc] Fix typo under colossalai and doc(#3618) 2023-04-26 11:38:43 +08:00
tensor_placement_policy.py [zero] reorganize zero/gemini folder structure (#3424) 2023-04-04 13:48:16 +08:00
tensor_utils.py fix typo with colossalai/trainer utils zero (#3908) 2023-06-07 16:08:37 +08:00