TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

may help numerical tests

Browse Source
This commit is contained in:
Mike J Innes 2018-02-19 12:51:02 +00:00
parent d22270050c
commit 4035745f6e
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/tracker/numeric.jl
View File

@ -19,4 +19,6 @@ function ngradient(f, xs::AbstractArray...)
return grads return grads
end end
gradcheck(f, xs...) = all(isapprox.(ngradient(f, xs...), gradient(f, xs...), rtol = 1e-5)) gradcheck(f, xs...) =
all(isapprox.(ngradient(f, xs...),
gradient(f, xs...), rtol = 1e-5, atol = 1e-5))