Update test/sde_neural.jl

ChrisRackauckas · web-flow · commit 9a9087d8ea26 · 2025-03-23T19:15:37.000-01:00
diff --git a/test/sde_neural.jl b/test/sde_neural.jl
@@ -196,7 +196,7 @@ end
     losses = []
     function callback(θ, state)
         begin
-            push!(losses, state.u)
+            push!(losses, l)
             if length(losses) % 1 == 0
                 println("Current loss after $(length(losses)) iterations: $(losses[end])")
             end
