Chengfeng Jia

icon company NTU icon location Singapore icon location I received my PhD from WHUT (China), was a visiting PhD student at BIASlab, TUE (Netherlands), and now work as a postdoc at NTU (Singapore).

Results 1 issues of Chengfeng Jia

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