ReverseDiff.jl
ReverseDiff.jl copied to clipboard
Get a error when calculating the gradient for LSTM
Hi, I build a network. ReverseDiff.gradient
works well for Dense layer. However, when I changed to a LSTM ReverseDiff.gradient(loss_mean, params),
it throw an error, MethodError: no method matching (::Flux.LSTMCell{ReverseDiff.TrackedArray{Float64, Float64, 2, Matrix{Float64}, Matrix{Float64}}, ReverseDiff.TrackedArray{Float64, Float64, 2, Matrix{Float64}, Matrix{Float64}}...