ColossalAI/tests/test_zero/test_gemini
Baizhou Zhang 0bb0b481b4 [gemini] fix argument naming during chunk configuration searching 2023-06-25 13:34:15 +08:00
..
test_chunk_mgrv2.py
test_chunkv2.py
test_fwd_bwd.py
test_gemini_use_rmt.py
test_get_torch_model.py
test_grad_clip.py
test_inference.py
test_optim.py
test_runtime_mem_tracer.py
test_search.py
test_zeroddp_state_dict.py
test_zeroddp_state_dict_shard.py
test_zerooptim_state_dict.py