geohotstan dd82b4c913 make onnx runner a class (#8647) ... * this * clean up * more clean ups and improve debug msg * more correct training toggler * remove manual training toggling * change some variable names * actually just add the training toggle for LIMIT envvar too * more refinement * __call__ and OnnxRunner * fix half pylint, other half is importing from onnx while this file is onnx.py, figure out later * ahhhh found another mistake * remove limit from __call__ --------- Co-authored-by: chenyu <chenyu@fastmail.com>		2025-01-20 10:11:05 -08: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	tests from grad uop path [pr] (#8313)	2024-12-18 09:25:05 -08: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	make onnx runner a class (#8647)	2025-01-20 10:11:05 -08:00
test_real_world.py	support multilb gradient [pr] (#8624)	2025-01-14 18:33:33 -08:00
test_resnet.py	Fix FC layer ResNet load_from_pretrained error (#8387)	2024-12-26 18:11:27 -05: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	move is_dtype_supported to device [pr] (#7575)	2024-11-07 20:38:03 +08:00