diff --git a/src/layers/recurrent.jl b/src/layers/recurrent.jl index eb227475..4707fc92 100644 --- a/src/layers/recurrent.jl +++ b/src/layers/recurrent.jl @@ -38,7 +38,7 @@ function (m::Recur)(xs...) return y end -treelike(Recur) +treelike(Recur, (:cell, :init)) Base.show(io::IO, m::Recur) = print(io, "Recur(", m.cell, ")")