final fixes

MartinuzziFrancesco · Nov 22, 2024 · b6dd982 · b6dd982
1 parent 9bfdf1e
commit b6dd982
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/peepholelstm_cell.jl b/src/peepholelstm_cell.jl
@@ -63,7 +63,7 @@ function PeepholeLSTMCell(
 end
 
 function (lstm::PeepholeLSTMCell)(inp::AbstractVecOrMat)
-    state = zeros_like(lstm, size(lstm.Wh, 2))
+    state = zeros_like(inp, size(lstm.Wh, 2))
     c_state = zeros_like(state)
     return lstm(inp, (state, c_state))
 end
@@ -75,7 +75,7 @@ function (lstm::PeepholeLSTMCell)(inp::AbstractVecOrMat,
     g = lstm.Wi * inp .+ lstm.Wh * c_state .+ b
     input, forget, cell, output = chunk(g, 4; dims = 1)
     new_cstate = @. sigmoid_fast(forget) * c_state + sigmoid_fast(input) * tanh_fast(cell)
-    new_state = @. sigmoid_fast(output) * tanh_fast(c′)
+    new_state = @. sigmoid_fast(output) * tanh_fast(new_cstate)
     return new_state, new_cstate
 end