From 099c9eb6bf2f5f90467feb7e496bf587f115dea0 Mon Sep 17 00:00:00 2001 From: Carlo Lucibello Date: Sat, 14 Sep 2024 04:24:38 +0200 Subject: [PATCH] naming --- GNNLux/src/layers/temporalconv.jl | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/GNNLux/src/layers/temporalconv.jl b/GNNLux/src/layers/temporalconv.jl index 69e784657..63c196a55 100644 --- a/GNNLux/src/layers/temporalconv.jl +++ b/GNNLux/src/layers/temporalconv.jl @@ -12,11 +12,11 @@ function (r::StatefulRecurrentCell)(g, x::AbstractMatrix, ps, st::NamedTuple) end function (r::StatefulRecurrentCell)(g, x::AbstractVector, ps, st::NamedTuple) - st, carry = st.cell, st.carry + stcell, carry = st.cell, st.carry for xᵢ in x - (out, carry), st = applyrecurrentcell(r.cell, g, xᵢ, ps.cell, st, carry) + (out, carry), stcell = applyrecurrentcell(r.cell, g, xᵢ, ps.cell, stcell, carry) end - return out, (; cell=st, carry) + return out, (; cell=stcell, carry) end function applyrecurrentcell(l, g, x, ps, st, carry)