Skip to content

Commit

Permalink
update BPINN_PDE_tests.jl
Browse files Browse the repository at this point in the history
  • Loading branch information
AstitvaAggarwal committed Nov 1, 2024
1 parent e298450 commit 0558db9
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions test/BPINN_PDE_tests.jl
Original file line number Diff line number Diff line change
Expand Up @@ -257,9 +257,10 @@ end
u = u .+ (u .* 0.2) .* randn(size(u))
dataset = [hcat(u, timepoints)]

# BPINNs are formulated with a mesh that must stay the same throughout sampling (as of now)
@testset "$(nameof(typeof(strategy)))" for strategy in [
StochasticTraining(200),
QuasiRandomTraining(200),
# StochasticTraining(200),
# QuasiRandomTraining(200),
GridTraining([0.02])
]
discretization = BayesianPINN([chainl], strategy; param_estim = true,
Expand Down

0 comments on commit 0558db9

Please sign in to comment.