botbw 80c3c8789b [Test/CI] remove test cases to reduce CI duration (#5753) ... * [test] smaller gpt2 test case * [test] reduce test cases: tests/test_zero/test_gemini/test_zeroddp_state_dict.py * [test] reduce test cases: tests/test_zero/test_gemini/test_grad_accum.py * [test] reduce test cases tests/test_zero/test_gemini/test_optim.py * Revert "[test] smaller gpt2 test case" Some tests might depend on the size of model (num of chunks) This reverts commit df705a5210. * [test] reduce test cases: tests/test_checkpoint_io/test_gemini_checkpoint_io.py * [CI] smaller test model for two mwo the two modifid cases * [CI] hardcode gpt model for tests/test_zero/test_gemini/test_search.py since we need a fixed answer there		2024-06-05 11:29:04 +08:00
..
test_gemini_checkpoint_io.py	[Test/CI] remove test cases to reduce CI duration (#5753)	2024-06-05 11:29:04 +08:00
test_gemini_torch_compability.py	[misc] Accelerate CI for zero and dist optim (#5758)	2024-06-05 11:25:19 +08:00
test_general_checkpoint_io.py	[hotfix] fix lr scheduler bug in torch 2.0 (#4864)	2023-10-12 14:04:24 +08:00
test_hybrid_parallel_plugin_checkpoint_io.py	[misc] Accelerate CI for zero and dist optim (#5758)	2024-06-05 11:25:19 +08:00
test_low_level_zero_checkpoint_io.py	[misc] Accelerate CI for zero and dist optim (#5758)	2024-06-05 11:25:19 +08:00
test_plugins_huggingface_compatibility.py	[misc] Accelerate CI for zero and dist optim (#5758)	2024-06-05 11:25:19 +08:00
test_torch_ddp_checkpoint_io.py	[misc] Accelerate CI for zero and dist optim (#5758)	2024-06-05 11:25:19 +08:00
test_torch_fsdp_checkpoint_io.py	[misc] refactor launch API and tensor constructor (#5666)	2024-04-29 10:40:11 +08:00
utils.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00