fix TRAINSET

SciML · Mar 19, 2024 · 2c87492 · 2c87492
1 parent fe8a819
commit 2c87492
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/NeuralPDE.jl b/src/NeuralPDE.jl
@@ -55,7 +55,7 @@ include("BPINN_ode.jl")
 include("PDE_BPINN.jl")
 include("dgm.jl")
 
-export NNODE, NNDAE, PINOODE
+export NNODE, NNDAE, PINOODE, TRAINSET
     PhysicsInformedNN, discretize,
     GridTraining, StochasticTraining, QuadratureTraining, QuasiRandomTraining,
     WeightedIntervalTraining,

diff --git a/test/PINO_ode_tests_gpu.jl b/test/PINO_ode_tests_gpu.jl
@@ -53,7 +53,7 @@ const gpud = gpu_device()
     ps = Lux.setup(Random.default_rng(), chain)[1] |> ComponentArray |> gpud
     opt = OptimizationOptimisers.Adam(0.03)
     alg = PINOODE(chain, opt, train_set; init_params = ps)
-    pino_solution = solve(prob, alg, verbose = true, maxiters = 1000)
+    pino_solution = solve(prob, alg, verbose = false, maxiters = 1000)
     predict = pino_solution.predict |> cpu
     ground = u_output_ |> cpu
     @test ground≈predict atol=1
@@ -102,7 +102,7 @@ end
     opt = OptimizationOptimisers.Adam(0.001)
     alg = PINOODE(
         flat_no, opt, train_set; init_params = ps, is_data_loss = true, is_physics_loss = true)
-    pino_solution = solve(prob, alg, verbose = true, maxiters = 1000)
+    pino_solution = solve(prob, alg, verbose = false, maxiters = 1000)
     predict = pino_solution.predict |> cpu
     ground = u_output_
     @test ground≈predict atol=2