diff --git a/src/layers/stateless.jl b/src/layers/stateless.jl index 0da3399c..4b868261 100644 --- a/src/layers/stateless.jl +++ b/src/layers/stateless.jl @@ -46,6 +46,6 @@ logitbinarycrossentropy(logŷ, y) = (1 - y)*logŷ - logσ(logŷ) """ function normalise(x::AbstractArray, dims::Int=1) μ′ = mean(x, dims = dims) - σ′ = std(x, dims = dims, mean = μ′) + σ′ = std(x, dims = dims, mean = μ′, corrected=false) return (x .- μ′) ./ σ′ end