Update test/sde_neural.jl

ChrisRackauckas · web-flow · commit b01a9d820344 · 2025-03-23T19:15:55.000-01:00
diff --git a/test/sde_neural.jl b/test/sde_neural.jl
@@ -116,7 +116,7 @@ Random.seed!(238248735)
     end
 
     losses = []
-    function callback(θ, state)
+    function callback(state, l)
         begin
             push!(losses, state.u)
             if length(losses) % 50 == 0
