tinygrad/test/models
chenyu a753c8e071
examples of new GPT2 and JIT change (#2261)
* var_vals are global

* working with global ish

* better

* fix export model

* fix tests

* better kv cache

* does it run?

* use where for kvmask

* fix excessive var_vals

* fix import

* how does multigpu use this?

* llama kinda work

* faster and simpler

* cleanup

* fix conversation mode

* test cleanups

* fix one more test

* test cleanup

---------

Co-authored-by: George Hotz <geohot@gmail.com>
2023-11-10 15:07:02 -05: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: make file transcription work, add basic CI test (#2042) 2023-10-13 17:13:35 -07:00
test_bert.py Torch eq fix (#1562) 2023-10-11 12:57:11 -07:00
test_efficientnet.py print test durations and add speed (#2107) 2023-10-18 13:46:42 -07:00
test_end2end.py clean up symlinking in benchmark (#2219) 2023-11-05 16:46:05 -08:00
test_mnist.py move state to nn/state (#1619) 2023-08-22 07:36:24 -07:00
test_onnx.py update openpilot supercombo.onnx to 0.9.4 (#1681) 2023-08-26 19:16:08 -04:00
test_real_world.py examples of new GPT2 and JIT change (#2261) 2023-11-10 15:07:02 -05:00
test_rnnt.py .cpu().numpy() -> .numpy() (#1594) 2023-08-21 09:53:29 -07:00
test_train.py print test durations and add speed (#2107) 2023-10-18 13:46:42 -07:00
test_waifu2x.py waifu2x vgg7: testcase, auto-RGBA->RGB, function to grab pretrained models, training "fix" (#2117) 2023-10-19 22:07:15 -07:00
test_whisper.py whisper: make file transcription work, add basic CI test (#2042) 2023-10-13 17:13:35 -07:00