ColossalAI/colossalai/zero/gemini
digger yu 9265f2d4d7
[NFC]fix typo colossalai/auto_parallel nn utils etc. (#3779)
* fix typo colossalai/autochunk auto_parallel amp

* fix typo colossalai/auto_parallel nn utils etc.
2023-05-23 15:28:20 +08:00
..
chunk [NFC]fix typo colossalai/auto_parallel nn utils etc. (#3779) 2023-05-23 15:28:20 +08:00
memory_tracer [NFC]fix typo colossalai/auto_parallel nn utils etc. (#3779) 2023-05-23 15:28:20 +08:00
__init__.py
colo_init_context.py [doc] Fix typo under colossalai and doc(#3618) 2023-04-26 11:38:43 +08:00
gemini_ddp.py [booster] gemini plugin support shard checkpoint (#3610) 2023-05-05 14:37:21 +08:00
gemini_hook.py
gemini_mgr.py
gemini_optimizer.py [misc] add verbose arg for zero and op builder (#3552) 2023-04-17 11:25:35 +08:00
placement_policy.py
utils.py