diff --git a/yklz/recurrent/lstm_peephole.py b/yklz/recurrent/lstm_peephole.py index 86dfce9..5a35c23 100644 --- a/yklz/recurrent/lstm_peephole.py +++ b/yklz/recurrent/lstm_peephole.py @@ -92,7 +92,7 @@ def step(self, inputs, states): o = self.recurrent_activation( x_o + K.dot(c_tm1 * rec_dp_mask[3], self.recurrent_kernel_o) ) - h = o * self.activation(c) + h = o * c if 0 < self.dropout + self.recurrent_dropout: h._uses_learning_phase = True return h, [h, c]