tinygrad/test
George Hotz 60df954377
Fix weight init: this work? (#391)
* this work?

* glorot uniform

* requies_grad broke

* propagate the None correctly

* so this weight init works

* ahh, i think it's this

* can't beat this

* glorot is best for ae

* remove comments
2022-09-25 16:46:33 -04:00
..
efficientnet add ViT test and car 2022-06-05 17:12:43 -07:00
__init__.py All devices are equal! (#196) 2020-12-15 23:44:08 -08:00
graph_batchnorm.py t.assign in optim 2022-08-20 14:04:33 -07:00
test_conv.py Fold reduce (#362) 2022-07-19 09:24:02 -07:00
test_efficientnet.py test_onnx works with enet also 2022-06-11 14:30:26 -07:00
test_gc.py set requires_grad to None (#387) 2022-09-21 11:16:02 -04:00
test_mnist.py Fix weight init: this work? (#391) 2022-09-25 16:46:33 -04:00
test_net_speed.py import tests from CL metal texture fix 2022-09-19 20:01:47 -04:00
test_nn.py add Linear to tinygrad.nn 2022-09-07 07:40:48 -07:00
test_onnx.py support float16 onnx weights (#384) 2022-09-15 09:12:18 -04:00
test_ops.py set requires_grad to None (#387) 2022-09-21 11:16:02 -04:00
test_optim.py tinygrad.nn (#367) 2022-08-18 07:41:00 -07:00
test_shapetracker.py tinygrad.nn (#367) 2022-08-18 07:41:00 -07:00
test_tensor.py set requires_grad to None (#387) 2022-09-21 11:16:02 -04:00
test_train.py import tests from CL metal texture fix 2022-09-19 20:01:47 -04:00