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}}...
Hi, I build a network.
ReverseDiff.gradient
works well for Dense layer. However, when I changed to a LSTMReverseDiff.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}}...