ColossalAI/tests/kit/model_zoo
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
..
custom [Feature] Distributed optimizers: Lamb, Galore, CAME and Adafactor (#5694) 2024-05-14 13:52:45 +08:00
diffusers [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
timm [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
torchaudio [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
torchrec [test] modify model supporting part of low_level_zero plugin (including correspoding docs) 2023-10-05 15:10:31 +08:00
torchvision [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
transformers [Test/CI] remove test cases to reduce CI duration (#5753) 2024-06-05 11:29:04 +08:00
__init__.py [devops] remove post commit ci (#5566) 2024-04-08 15:09:40 +08:00
executor.py [test] merge old components to test to model zoo (#4945) 2023-10-20 10:35:08 +08:00
registry.py [devops] remove post commit ci (#5566) 2024-04-08 15:09:40 +08:00