ColossalAI/colossalai/zero/sharded_model
ver217 e3fde4ee6b
fix import error in sharded model v2 (#1053)
2022-06-02 13:48:22 +08:00
..
__init__.py
_utils.py [gemini] add GeminiMemoryManger (#832) 2022-04-24 13:08:48 +08:00
reduce_scatter.py
sharded_model_v2.py fix import error in sharded model v2 (#1053) 2022-06-02 13:48:22 +08:00
utils.py