-
-
Notifications
You must be signed in to change notification settings - Fork 608
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Adding return state option to recurrent layers #2557
Conversation
tests are failing |
is this something that has to be addressed at the Functor level? the fail is at https://github.com/FluxML/Flux.jl/blob/master/test/test_utils.jl#L66 with
so I guess it's trying to rebuild the struct but it can't since it doesn't have the return_state parametrization information. I can add a workaround like RNN(rnn::RNNCell) = RNN{false, RNNCell}(rnn) but this wouldn't solve it at the level that it's needed. The tests would still fail for |
yes this needs a custom Functors.functor(rnn::RNN{S}) = (; rnn.cell), ch -> RNN{S}(ch.cell) (not tested) |
Codecov ReportAll modified and coverable lines are covered by tests ✅
Additional details and impacted files@@ Coverage Diff @@
## master #2557 +/- ##
===========================================
+ Coverage 32.50% 65.53% +33.02%
===========================================
Files 34 34
Lines 2003 2083 +80
===========================================
+ Hits 651 1365 +714
+ Misses 1352 718 -634 ☔ View full report in Codecov by Sentry. |
Addressing the
return_state
as discussed in #2514.PR Checklist
Entry in NEWS.md