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

Error statement

Browse Source
This commit is contained in:
Avik Pal 2018-07-17 10:14:23 +05:30
parent 4035641f00
commit 531ecccd38
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/cuda/cudnn.jl
View File

@ -189,7 +189,5 @@ batchnorm(g::TrackedArray, b::TrackedArray, x::CuArray{T}, running_mean::CuArray
@grad function batchnorm(g, b, x, running_mean, running_var, momentum; kw...)
y = batchnorm(data(g), data(b), data(x), running_mean, running_var, momentum; kw...)
deriv_tup = ∇batchnorm(data(g), data(b), data(x), Δ, running_mean, running_var, momentum,
cache = cache, alpha = alpha, beta = beta, eps = eps, training = training)
y, Δ -> (nobacksies(:batchnorm, ∇batchnorm(data.(g, b, x, Δ), running_mean, running_var, momentum; kw...)), nothing, nothing, nothing)
end