dustcollector12andGitHub 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