tinygrad/test
dustcollector12 ee99d016e9
tensor implementation for rmsprop and adam (#121)
* tensor implementation for rmsprop and adam

* test_mnist.py extended to cover sgd, rmsprop and adam on cpu and gpu

* number of steps reduced for adam from 1000 to 200
2020-11-16 15:07:49 -08:00
..
test_mnist.py tensor implementation for rmsprop and adam (#121) 2020-11-16 15:07:49 -08:00
test_net_speed.py GPU MaxPool2D.backward(); TinyConvNet train passes (#103) 2020-11-11 07:58:43 -08:00
test_nn.py unit test for batchnorm2d 2020-10-30 08:19:58 -07:00
test_ops.py unbroadcasting works on the GPU 2020-11-16 09:16:55 -08:00
test_optim.py efficient version of adam (#20) 2020-10-27 15:54:40 -07:00
test_tensor.py move gradcheck to extra, clean up unbroadcast 2020-11-16 08:03:31 -08:00