diff --git a/src/layers/normalise.jl b/src/layers/normalise.jl index 728c91df..97e88d81 100644 --- a/src/layers/normalise.jl +++ b/src/layers/normalise.jl @@ -311,7 +311,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)), + GroupNorm(G, λ, initβ(chs), initγ(chs), zeros(G,1), ones(G,1), ϵ, momentum) function(gn::GroupNorm)(x)