mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity
tinygrad/test/models
History
JaSpa99 d3d58a37e5
 Bert: use Tensor.scaled_dot_product_attention (#1528) 
* use scaled attn from Tensor

* add a test for bert

* linter

* no more tokenizer

* without loading weights

* remove prints

* tribute to linter lords

* smaller input and less runs

* small bert
2023-08-12 08:46:04 -07:00
..
efficientnet test cleanups 2023-03-03 06:36:06 -08:00
test_bert.py Bert: use Tensor.scaled_dot_product_attention (#1528) 2023-08-12 08:46:04 -07:00
test_efficientnet.py Inference test: add tests for ResNet50 (#773) 2023-05-13 21:18:15 -07:00
test_end2end.py CI < 5 minutes (#1252) 2023-07-23 13:00:56 -07:00
test_mnist.py CI < 5 minutes (#1252) 2023-07-23 13:00:56 -07:00
test_onnx.py CI < 5 minutes (#1252) 2023-07-23 13:00:56 -07:00
test_real_world.py Raise memory limit for CIFAR test (#1499) 2023-08-08 19:40:56 -04:00
test_rnnt.py make tests faster + add onnx (#815) 2023-05-27 08:53:32 -07:00
test_train.py CI < 5 minutes (#1252) 2023-07-23 13:00:56 -07:00