Flux RNN training giving `BoundsError`

This is BoundsError calling Flux.reset! · Issue #1297 · FluxML/Zygote.jl · GitHub , fix from there is:

using ChainRulesCore, Zygote
ChainRulesCore.@non_differentiable foreach(f, ::Tuple{})
Zygote.refresh()

Could surely be added here.

2 Likes