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

brackets corrected

This commit is contained in:
Manjunath Bhat 2019-07-12 21:03:57 +05:30 committed by GitHub
parent fc1c0d58ed
commit 4ef5ec0005
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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.β])