github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/zero/gemini
History
Hongxin Liu a39a5c66fe
Merge branch 'main' into feature/shardformer 		2023-09-04 23:43:13 +08:00
..
chunk [gemini] improve compatibility and add static placement policy (#4479) 2023-08-24 09:29:25 +08:00
memory_tracer [gemini] improve compatibility and add static placement policy (#4479) 2023-08-24 09:29:25 +08:00
__init__.py [gemini] improve compatibility and add static placement policy (#4479) 2023-08-24 09:29:25 +08:00
colo_init_context.py [doc] Fix typo under colossalai and doc(#3618) 2023-04-26 11:38:43 +08:00
gemini_ddp.py Merge branch 'main' into feature/shardformer 2023-09-04 23:43:13 +08:00
gemini_hook.py
…
gemini_mgr.py [gemini] improve compatibility and add static placement policy (#4479) 2023-08-24 09:29:25 +08:00
gemini_optimizer.py Merge branch 'main' into feature/shardformer 2023-09-04 23:43:13 +08:00
placement_policy.py [gemini] improve compatibility and add static placement policy (#4479) 2023-08-24 09:29:25 +08:00
utils.py [gemini] improve compatibility and add static placement policy (#4479) 2023-08-24 09:29:25 +08:00