ReverseDiff.jl icon indicating copy to clipboard operation
ReverseDiff.jl copied to clipboard

Get a error when calculating the gradient for LSTM

Open Chengfeng-Jia opened this issue 2 years ago • 0 comments

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

Chengfeng-Jia avatar Dec 04 '22 13:12 Chengfeng-Jia