5 Commits

Author	SHA1	Message	Date
chenyu	25091951ba	update test/models (#12142) ... minor fix and run more stuff in tinygrad for speed	2025-09-12 16:43:28 -04:00
George Hotz	0cbf6c1811	move things, clean up extra (#2292) ... * move things * idk why pylint needs that now * delete unused	2023-11-13 20:18:40 -08:00
Yixiang Gao	8d6662a741	.cpu().numpy() -> .numpy() (#1594) ... * .cpu().numpy() -> .numpy() * restore ops_torch * restore test_speed_v_torch	2023-08-21 09:53:29 -07:00
George Hotz	a3feee29c5	make tests faster + add onnx (#815) ... * search one dir, disable slow * onnx tests * fast rnnt test	2023-05-27 08:53:32 -07:00
wozeparrot	01ae45a43c	Add mlperf RNN-T model (#782) ... * feat: initial rnn-t * feat: working with BS>1 * feat: add lstm test * feat: test passing hidden * clean: cleanup * feat: specify start * feat: way faster lstm & model * fix: default batch size * feat: optimization * fix: fix metrics * fix: fix feature splicing * feat: cleaner stacktime * clean: remove unused import * clean: remove extra prints * fix: fix tests and happy llvm * feat: have the librispeech dataset in its own dir * clean: unused variable * feat: no longer need numpy for the embedding + slightly more memory efficient lstm * fix: forgot to remove something that broke tests * feat: use relative paths * feat: even faster * feat: remove pointless transposes in StackTime * fix: correct forward * feat: switch to soundfile for loading and fix some leaks * feat: add comment about initial dataset setup * feat: jit more things * feat: default batch size back to 1 larger than 1 is broken again :( and even in the reference implementation it gives worse results	2023-05-25 00:41:21 -07:00