From 4916c8e6da46d13078e4bfac6f10312a3fe44ce8 Mon Sep 17 00:00:00 2001 From: Avik Pal Date: Wed, 27 Jun 2018 14:54:49 +0530 Subject: [PATCH] Add treelike for now --- src/layers/normalise.jl | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/src/layers/normalise.jl b/src/layers/normalise.jl index 25832c07..e43c76b7 100644 --- a/src/layers/normalise.jl +++ b/src/layers/normalise.jl @@ -144,11 +144,13 @@ end (BN::BatchNorm)(x) = BN.λ.(batchnorm(BN.γ, BN.β, x, BN.μ, BN.σ, BN.momentum; cache = BN.cache, alpha = 1, beta = 0, eps = BN.ϵ, training = BN.active)) -children(BN::BatchNorm) = - (BN.λ, BN.β, BN.γ, BN.μ, BN.σ, BN.ϵ, BN.momentum, BN.active) +Flux.treelike(BatchNorm) -mapchildren(f, BN::BatchNorm) = # e.g. mapchildren(cu, BN) - BatchNorm(BN.λ, f(BN.β), f(BN.γ), f(BN.μ), f(BN.σ), BN.ϵ, BN.momentum, BN.active) +# children(BN::BatchNorm) = +# (BN.λ, BN.β, BN.γ, BN.μ, BN.σ, BN.ϵ, BN.momentum, BN.active) +# +# mapchildren(f, BN::BatchNorm) = # e.g. mapchildren(cu, BN) +# BatchNorm(BN.λ, f(BN.β), f(BN.γ), f(BN.μ), f(BN.σ), BN.ϵ, BN.momentum, BN.active) _testmode!(BN::BatchNorm, test) = (BN.active = !test)