Ryan Neph 5265f6c578 add AvgPool2D backward pass on GPU (#68)

2020-11-07 12:27:29 -08:00

..

__init__.py

factor out a couple nn ops

2020-10-29 08:01:12 -07:00

gradcheck.py

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

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

nn.py

some broadcasting, pool test is fail

2020-11-07 11:29:42 -08:00

ops.py

weee, logsoftmax on GPU. very slow

2020-11-07 07:59:37 -08:00

opsgpu.py

add AvgPool2D backward pass on GPU (#68)

2020-11-07 12:27:29 -08:00

optim.py

gpu SGD runs now

2020-11-02 07:19:12 -08:00

tensor.py

fix strided convs, GPU env var for enet

2020-11-07 10:26:37 -08:00

utils.py

fetch() 'ran out of input' fix (#65)

2020-11-07 09:45:10 -08:00