I am optimizing a simple ODE problem with DiffEqFlux to get started with Julia. This code worked on TensorFlow. In the following code, running main(S=5, B=256) works, but main(S=7, B=256) fails. I previously used EnsembleProblem with same code and got same results. The error begins: ERROR: Need an…

DiffEqFlux gradient error: breaks with increasing size of problem, size should not affect autodiff

ChrisRackauckas November 17, 2021, 12:32am 8

Looks like a bug in the map adjoint.

Topic		Replies	Views
DiffEqFlux / diffeq_adjoint: no method matching back!(::Float64) Machine Learning diffeq	3	1141	May 28, 2019
DiffEqFlux and Lux/Flux General Usage	7	1112	January 27, 2023
DiffEqParamEstim.jl on Ensemble problems: autodiff doesn't work Modelling & Simulations	2	721	May 23, 2020
Error in diffeq_fd command of DiffEqFlux Machine Learning diffeq , flux	4	518	June 29, 2019
Different gradient behaviour depending on ODE-Definition New to Julia question , diffeq , diffeqflux	11	592	January 23, 2022