tinygrad/test/models
kormann f5dd25d376
enable whisper batch for long sequences (#6458)
* long batch +test

* long batch +test

* cleanup

* rollback syntactic changes

---------

Co-authored-by: chenyu <chenyu@fastmail.com>
2024-09-17 00:42:10 -04:00
..
efficientnet test cleanups 2023-03-03 06:36:06 -08:00
waifu2x waifu2x vgg7: testcase, auto-RGBA->RGB, function to grab pretrained models, training "fix" (#2117) 2023-10-19 22:07:15 -07:00
whisper whisper: support batch inference, add librispeech WER test (#2074) 2023-11-16 13:50:08 -08:00
__init__.py more test cleanups (#2631) 2023-12-05 16:17:57 -08:00
test_bert.py Apply ruff linting rules to tests (#2473) 2023-11-27 21:24:06 -08:00
test_efficientnet.py db in wal mode (#5388) 2024-07-12 20:43:36 -07:00
test_end2end.py add failing regression test for image (#5540) 2024-07-17 17:27:18 -07:00
test_mnist.py skip a few clang test that took > 30 seconds in CI (#4126) 2024-04-10 02:00:34 -04:00
test_onnx.py fix some long lines in tests (#3006) 2024-01-03 23:53:33 -05:00
test_real_world.py skip some redundant and slow tests in ci (#5416) 2024-07-12 14:43:13 -04:00
test_resnet.py update resnet.load_from_pretrained (#5040) 2024-06-18 16:29:22 -04:00
test_rnnt.py move things, clean up extra (#2292) 2023-11-13 20:18:40 -08:00
test_train.py Remove pytest markers (#2831) 2023-12-18 18:53:28 -05:00
test_waifu2x.py Apply ruff linting rules to tests (#2473) 2023-11-27 21:24:06 -08:00
test_whisper.py enable whisper batch for long sequences (#6458) 2024-09-17 00:42:10 -04:00