![]() 957: Add some gradient checking tests on GPUs r=dhairyagandhi96 a=dhairyagandhi96 Good to add generic tests for tracking gradients through the various layers on the GPU. Co-authored-by: Dhairya Gandhi <dhairya@juliacopmuting.com> Co-authored-by: Dhairya Gandhi <dhairya@juliacomputing.com> |
||
---|---|---|
.. | ||
cuda | ||
layers | ||
data.jl | ||
onehot.jl | ||
optimise.jl | ||
runtests.jl | ||
utils.jl |