adamritter 5ea3d76dfb Topological sort, zero_grads (#119) ...

* Topological sort, zero_grads

* Bug fix, add test

* Add zero_grads

* Put deepwalk function in backward

* Move zero_grad to optim

* Fix gradcheck hack

Co-authored-by: holonomicjl <58403584+holonomicjl@users.noreply.github.com>

2020-11-15 20:25:29 -08:00

..

test_mnist.py

Topological sort, zero_grads (#119)

2020-11-15 20:25:29 -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

Topological sort, zero_grads (#119)

2020-11-15 20:25:29 -08:00

test_optim.py

efficient version of adam (#20)

2020-10-27 15:54:40 -07:00

test_tensor.py

Add GPU Support! (do not merge yet) (#41)

2020-11-01 07:00:49 -08:00