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

Merge branch 'zygote' of https://github.com/FluxML/Flux.jl into zygote

Browse Source
This commit is contained in:
thebhatman 2019-07-12 22:20:34 +05:30
commit d0b94b88f6
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
test/cuda/cudnn.jl
View File

@ -15,7 +15,7 @@ trainmode(f, x...) = forward(f, x...)[1]
@test cpu(data(cy)) ≈ data(y)
g = gradient(()->sum(m(x)), params(m))
cg = gradient(()->sum(cm(cx), params(cm))
cg = gradient(()->sum(cm(cx)), params(cm))
@test g.grads[m.γ] ≈ cpu(cg.grads[cm.γ])
@test g.grads[m.β] ≈ cpu(cg.grads[cm.β])
@ -35,7 +35,7 @@ trainmode(f, x...) = forward(f, x...)[1]
@test cpu(data(cy)) ≈ data(y)
g = gradient(()->sum(m(x)), params(m))
cg = gradient(()->sum(cm(cx), params(cm))
cg = gradient(()->sum(cm(cx)), params(cm))
@test g.grads[m.γ] ≈ cpu(cg.grads[cm.γ])
@test g.grads[m.β] ≈ cpu(cg.grads[cm.β])