George Hotz 866b759d3b match torch api for pad2d

2020-11-09 17:48:56 -08:00

..

test_mnist.py

getting convs to work on gpu

2020-11-07 09:17:57 -08:00

test_net_speed.py

fix tests

2020-10-29 08:19:07 -07:00

test_nn.py

unit test for batchnorm2d

2020-10-30 08:19:58 -07:00

test_ops.py

match torch api for pad2d

2020-11-09 17:48:56 -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