From dfd2965e85fab02589874a7db387b3b5aa92481e Mon Sep 17 00:00:00 2001 From: thebhatman Date: Tue, 11 Jun 2019 22:32:54 +0530 Subject: [PATCH] GroupNorm tests corrected --- 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 d02aee35..01817948 100644 --- a/src/layers/normalise.jl +++ b/src/layers/normalise.jl @@ -300,7 +300,7 @@ end GroupNorm(chs::Integer, G::Integer, λ = identity; initβ = (i) -> zeros(Float32, i), initγ = (i) -> ones(Float32, i), ϵ = 1f-5, momentum = 0.1f0) = GroupNorm(G, λ, param(initβ(chs)), param(initγ(chs)), - zeros(G,1), ones(G,1), ϵ, momentum, true) + zeros(G,1), ones(G,1), ϵ, momentum) function(gn::GroupNorm)(x) size(x,ndims(x)-1) == length(gn.β) || error("Group Norm expected $(length(gn.β)) channels, but got $(size(x,ndims(x)-1)) channels")