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.
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.