gru_reset

jcvasquezc · Jul 13, 2020 · 68c6d40 · 68c6d40
1 parent 9f62e92
commit 68c6d40
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/phonet/phonet.py b/phonet/phonet.py
@@ -143,8 +143,8 @@ def model(self, input_size):
         input_data=Input(shape=(input_size))
         x=input_data
         x=BatchNormalization()(x)
-        x=Bidirectional(GRU(self.GRU_size, recurrent_dropout=self.recurrent_droput_prob, return_sequences=True))(x)
-        x=Bidirectional(GRU(self.GRU_size, recurrent_dropout=self.recurrent_droput_prob, return_sequences=True))(x)
+        x=Bidirectional(GRU(self.GRU_size, recurrent_dropout=self.recurrent_droput_prob, return_sequences=True, reset_after=False))(x)
+        x=Bidirectional(GRU(self.GRU_size, recurrent_dropout=self.recurrent_droput_prob, return_sequences=True, reset_after=False))(x)
         x=Dropout(0.2)(x)
         x = TimeDistributed(Dense(self.hidden_size, activation='relu'))(x)
         x=Dropout(0.2)(x)