fix

impICNF · Sep 7, 2023 · c630b31 · c630b31
1 parent fd86171
commit c630b31
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/src/cond_rnode.jl b/src/cond_rnode.jl
@@ -57,8 +57,12 @@ function construct(
     autodiff_backend::ADTypes.AbstractADType = AutoZygote(),
     sol_args::Tuple = (),
     sol_kwargs::Dict = Dict(
-        :alg_hints => [:nonstiff, :memorybound],
-        :reltol => 1e-2 + eps(1e-2),
+        :alg_hints => [:nonstiff],
+        :alg => VCABM(),
+        :sensealg => InterpolatingAdjoint(; autodiff = true, autojacvec = ZygoteVJP()),
+        :reltol => sqrt(eps(one(Float32))),
+        :abstol => eps(one(Float32)),
+        :maxiters => typemax(Int32),
     ),
     rng::AbstractRNG = Random.default_rng(),
     λ₁::AbstractFloat = convert(data_type, 1e-2),

diff --git a/src/rnode.jl b/src/rnode.jl
@@ -59,8 +59,12 @@ function construct(
     autodiff_backend::ADTypes.AbstractADType = AutoZygote(),
     sol_args::Tuple = (),
     sol_kwargs::Dict = Dict(
-        :alg_hints => [:nonstiff, :memorybound],
-        :reltol => 1e-2 + eps(1e-2),
+        :alg_hints => [:nonstiff],
+        :alg => VCABM(),
+        :sensealg => InterpolatingAdjoint(; autodiff = true, autojacvec = ZygoteVJP()),
+        :reltol => sqrt(eps(one(Float32))),
+        :abstol => eps(one(Float32)),
+        :maxiters => typemax(Int32),
     ),
     rng::AbstractRNG = Random.default_rng(),
     λ₁::AbstractFloat = convert(data_type, 1e-2),