Update test/sde_neural.jl

ChrisRackauckas · web-flow · commit 4ab5bb45b6d2 · 2025-03-23T19:15:42.000-01:00
diff --git a/test/sde_neural.jl b/test/sde_neural.jl
@@ -194,7 +194,7 @@ end
 
     # Actually training/fitting the model
     losses = []
-    function callback(θ, state)
+    function callback(state, l)
         begin
             push!(losses, l)
             if length(losses) % 1 == 0
