From b9a66c679d32b633ec1cf280af9a610ec626dee9 Mon Sep 17 00:00:00 2001 From: Brad Safnuk Date: Thu, 22 Mar 2018 22:20:21 -0400 Subject: [PATCH] =?UTF-8?q?Fix=20error=20in=20initialization=20of=20=CF=83?= =?UTF-8?q?.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/layers/normalisation.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/layers/normalisation.jl b/src/layers/normalisation.jl index e8901185..90649a93 100644 --- a/src/layers/normalisation.jl +++ b/src/layers/normalisation.jl @@ -114,7 +114,7 @@ end BatchNorm(dims::Integer...; λ = identity, initβ = zeros, initγ = ones, ϵ = 1e-8, momentum = .1) = -BatchNorm(λ, param(initβ(dims)), param(initγ(dims)), zeros(dims), zeros(dims), ϵ, momentum, true) +BatchNorm(λ, param(initβ(dims)), param(initγ(dims)), zeros(dims), ones(dims), ϵ, momentum, true) function (BN::BatchNorm)(x) λ, γ, β = BN.λ, BN.γ, BN.β