Skip to content

Commit 7eb3b45

Browse files
committed
test via zygote
1 parent 261ec38 commit 7eb3b45

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

src/base_icnf.jl

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,16 @@ function construct(
1818
FillArrays.Zeros{data_type}(nvars + naugmented),
1919
FillArrays.Eye{data_type}(nvars + naugmented),
2020
),
21-
sol_kwargs::NamedTuple = (;),
21+
sol_kwargs::NamedTuple = (;
22+
sensealg = SciMLSensitivity.QuadratureAdjoint(;
23+
autodiff = true,
24+
autojacvec = SciMLSensitivity.ZygoteVJP(),
25+
),
26+
save_everystep = false,
27+
reltol = sqrt(eps(one(Float32))),
28+
abstol = eps(one(Float32)),
29+
maxiters = typemax(Int32),
30+
),
2231
rng::Random.AbstractRNG = rng_AT(resource),
2332
λ₁::AbstractFloat = if aicnf <: Union{RNODE, CondRNODE}
2433
convert(data_type, 1.0e-2)

0 commit comments

Comments
 (0)