diff --git a/src/losses/functions.jl b/src/losses/functions.jl index a4db0de7c3..ff5f23a715 100644 --- a/src/losses/functions.jl +++ b/src/losses/functions.jl @@ -44,8 +44,7 @@ julia> Flux.mse(y_model, y_true) """ function mse(ŷ, y; agg = mean) _check_sizes(ŷ, y) - error = ŷ .- y - real(agg(error .* conj(error))) + agg(abs2.(ŷ .- y)) end """