You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Last Zygote version makes the following test (which lives in test/cuda/curnn.jl, altough the problem is not cuda-specific) fail
using Flux, Test
m =RNN(10, 5)
x =rand(10)
dm =gradient(m ->sum(m(x)), m)[1]
Flux.reset!(m)
θ =gradient(() ->sum(m(x)), params(m))
@testcollect(dm[].cell[].Wi) ==collect(θ[m.cell.Wi])
The problem is that structural gradient dm doesn't contain all of its fields' gradients anymore, while the Params derivative seems fine
Last Zygote version makes the following test (which lives in test/cuda/curnn.jl, altough the problem is not cuda-specific) fail
The problem is that structural gradient
dm
doesn't contain all of its fields' gradients anymore, while the Params derivative seems fine@DhairyaLGandhi This is causing the CI failures observed in #1204
The text was updated successfully, but these errors were encountered: