Fix residual

SciML · Nov 6, 2023 · f2d3285 · f2d3285
1 parent 87203f4
commit f2d3285
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/nonlinear_least_squares.jl b/test/nonlinear_least_squares.jl
@@ -12,12 +12,12 @@ y_target = true_function(x, θ_true)
 
 function loss_function(θ, p)
     ŷ = true_function(p, θ)
-    return abs2.(ŷ .- y_target)
+    return ŷ .- y_target
 end
 
 function loss_function(resid, θ, p)
     true_function(resid, p, θ)
-    resid .= abs2.(resid .- y_target)
+    resid .= resid .- y_target
     return resid
 end