From b6609f9eb9a2d7c13640e19a9935ebfccd5429e9 Mon Sep 17 00:00:00 2001 From: KirillZubov Date: Tue, 29 Oct 2024 20:18:55 +0400 Subject: [PATCH] fix --- test/PINO_ode_tests.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/PINO_ode_tests.jl b/test/PINO_ode_tests.jl index 5305e03da..ff33ce176 100644 --- a/test/PINO_ode_tests.jl +++ b/test/PINO_ode_tests.jl @@ -141,8 +141,8 @@ end #generate data ground_analytic = (u0, p, t) -> u0 + p * t^3 / 3 - sol = ground_analytic.(u0, p, vec(t)) p, t = get_trainset(deeponet, bounds, number_of_parameters, tspan, dt) + sol = ground_analytic.(u0, p, vec(t)) function additional_loss_(phi, θ) u = phi((p, t), θ) norm = prod(size(u)) @@ -152,7 +152,7 @@ end alg = PINOODE( deeponet, opt, bounds, number_of_parameters; strategy = strategy, additional_loss = additional_loss_) - sol = solve(prob, alg, verbose = true, maxiters = 3000) + sol = solve(prob, alg, verbose = false, maxiters = 3000) p, t = get_trainset(deeponet, bounds, number_of_parameters, tspan, dt) ground_solution = ground_analytic.(u0, p, vec(t))