nicer batchnorm shape error

2018-04-15 20:29:25 +01:00 · 2018-04-15 20:29:25 +01:00 · d12fb98f2a
commit d12fb98f2a
parent 8f29968c32
1 changed files with 2 additions and 0 deletions
--- a/src/layers/normalise.jl
+++ b/src/layers/normalise.jl
@ -114,6 +114,8 @@ BatchNorm(chs::Integer, λ = identity;
            zeros(chs), ones(chs), ϵ, momentum, true)

 function (BN::BatchNorm)(x)
+  size(x, ndims(x)-1) == length(BN.β) ||
+    error("BatchNorm expected $(length(BN.β)) channels, got $(size(x, ndims(x)-1))")
  λ, γ, β = BN.λ, BN.γ, BN.β
  dims = length(size(x))
  channels = size(x, dims-1)