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