1
0
Fork 0
tinygrab/test/models
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
..
efficientnet test cleanups 2023-03-03 06:36:06 -08:00
test_efficientnet.py Inference test: add tests for ResNet50 (#773) 2023-05-13 21:18:15 -07:00
test_end2end.py fix batchnorm at training (#753) 2023-04-19 08:01:04 -07:00
test_mnist.py remove RMSprop, nobody uses it anymore 2023-03-20 12:31:34 -07:00
test_onnx.py fix external test + speed 2023-03-03 06:46:16 -08:00
test_rnnt.py Add mlperf RNN-T model (#782) 2023-05-25 00:41:21 -07:00
test_train.py ConvNeXt + extras (#657) 2023-03-06 22:10:56 -08:00