diff --git a/seq2seq/cells.py b/seq2seq/cells.py index 1f89070..156a118 100644 --- a/seq2seq/cells.py +++ b/seq2seq/cells.py @@ -97,7 +97,7 @@ def build_model(self, input_shape): z0, z1, z2, z3 = get_slices(z, 4) i = Activation(self.recurrent_activation)(z0) - f = Activation(self.recurrent_activation)(z0) + f = Activation(self.recurrent_activation)(z1) c = add([multiply([f, c_tm1]), multiply([i, Activation(self.activation)(z2)])]) o = Activation(self.recurrent_activation)(z3)