Yanjia0 c614a99d28 [NFC] polish colossalai/auto_parallel/offload/amp_optimizer.py code style (#4255)		2023-07-26 14:12:57 +08:00
..
__init__.py	[auto-parallel] add auto-offload feature (#3154)	2023-03-21 14:17:41 +08:00
amp_optimizer.py	[NFC] polish colossalai/auto_parallel/offload/amp_optimizer.py code style (#4255)	2023-07-26 14:12:57 +08:00
base_offload_module.py	[zero] reorganize zero/gemini folder structure (#3424)	2023-04-04 13:48:16 +08:00
mem_optimize.py	[test] fixed gemini plugin test (#3411)	2023-04-03 17:12:22 +08:00
region.py	[doc] Fix typo under colossalai and doc(#3618)	2023-04-26 11:38:43 +08:00
region_manager.py	[auto-parallel] add auto-offload feature (#3154)	2023-03-21 14:17:41 +08:00
runtime.py	[test] fixed gemini plugin test (#3411)	2023-04-03 17:12:22 +08:00
solver.py	[auto-parallel] add auto-offload feature (#3154)	2023-03-21 14:17:41 +08:00
training_simulator.py	[doc] Fix typo under colossalai and doc(#3618)	2023-04-26 11:38:43 +08:00
util.py	[test] fixed gemini plugin test (#3411)	2023-04-03 17:12:22 +08:00