Skip to content

Commit

Permalink
don't use such a high tolerance
Browse files Browse the repository at this point in the history
  • Loading branch information
ChrisRackauckas committed Jun 26, 2021
1 parent 8a27810 commit 65601fe
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion Project.toml
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
name = "DiffEqFlux"
uuid = "aae7a2af-3d4f-5e19-a356-7da93b79d9d0"
authors = ["Chris Rackauckas <[email protected]>"]
version = "1.40.0"
version = "1.40.1"

[deps]
Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
Expand Down
4 changes: 2 additions & 2 deletions test/multiple_shoot.jl
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
using DiffEqFlux, DiffEqSensitivity, OrdinaryDiffEq, Flux, Optim, Test
using DiffEqFlux, DiffEqSensitivity, OrdinaryDiffEq, Test
using DiffEqFlux: group_ranges

## Test group partitioning helper function
Expand Down Expand Up @@ -61,7 +61,7 @@ continuity_term = 200
function loss_multiple_shooting(p)
return multiple_shoot(p, ode_data, tsteps, prob_node, loss_function, Tsit5(),
group_size; continuity_term,
abstol=1e-3, reltol=1e-3) # test solver kwargs
abstol=1e-8, reltol=1e-6) # test solver kwargs
end

res_ms = DiffEqFlux.sciml_train(loss_multiple_shooting, neuralode.p,
Expand Down

0 comments on commit 65601fe

Please sign in to comment.