From 681d8c4dfcafe311024425d85ab846d6ed89c251 Mon Sep 17 00:00:00 2001 From: Avik Pal Date: Thu, 28 Jun 2018 12:11:32 +0530 Subject: [PATCH] Remove cache --- src/layers/normalise.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/layers/normalise.jl b/src/layers/normalise.jl index 04082a73..8d2c3ffd 100644 --- a/src/layers/normalise.jl +++ b/src/layers/normalise.jl @@ -110,7 +110,7 @@ end BatchNorm(chs::Integer, λ = identity; initβ = zeros, initγ = ones, ϵ = 1e-5, momentum = .1) = BatchNorm(λ, param(initβ(chs)), param(initγ(chs)), - zeros(chs), ones(chs), ϵ, momentum, nothing, true) + zeros(chs), ones(chs), ϵ, momentum, true) function (BN::BatchNorm)(x) size(x, ndims(x)-1) == length(BN.β) ||