You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/tests/kit/model_zoo
flybird11111 29695cf70c
[example]add gpt2 benchmark example script. (#5295)
9 months ago
..
custom [test] merge old components to test to model zoo (#4945) 1 year ago
diffusers [misc] update pre-commit and run all files (#4752) 1 year ago
timm [misc] update pre-commit and run all files (#4752) 1 year ago
torchaudio [misc] update pre-commit and run all files (#4752) 1 year ago
torchrec [test] modify model supporting part of low_level_zero plugin (including correspoding docs) 1 year ago
torchvision [misc] update pre-commit and run all files (#4752) 1 year ago
transformers [workflow] fixed oom tests (#5275) 11 months ago
__init__.py [workflow] fixed build CI (#5240) 11 months ago
executor.py [test] merge old components to test to model zoo (#4945) 1 year ago
registry.py [example]add gpt2 benchmark example script. (#5295) 9 months ago