fixup! test: update NNODE tests - forward pass in additional loss

SciML · Jun 29, 2024 · dcf9007 · dcf9007
1 parent ba1ca23
commit dcf9007
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/NNODE_tests.jl b/test/NNODE_tests.jl
@@ -215,7 +215,7 @@ end
         luxchain = Lux.Chain(Lux.Dense(1, 5, Lux.σ), Lux.Dense(5, 1))
         (u_, t_) = (u_analytical(ts), ts)
         function additional_loss(phi, θ)
-            return sum(sum(abs2, vec(phi(t_, θ)) .- u_)) / length(u_) 
+            return sum(sum(abs2, vec(phi(t_, θ)) .- u_)) / length(u_)
         end
         alg1 = NNODE(luxchain, opt, strategy = StochasticTraining(1000),
             additional_loss = additional_loss)