No gradient is calculated during Flux training with ODE

The solution to this is the same as The tracked parameters are not updated during Flux.training - #2 by ChrisRackauckas

1 Like