diff --git a/speech_model.py b/speech_model.py index 7d81df3..5fe9b70 100644 --- a/speech_model.py +++ b/speech_model.py @@ -17,12 +17,12 @@ def create_base_rnn_network(input_dim): ''' inp = Input(shape=input_dim) # ls0 = LSTM(512, return_sequences=True)(inp) - ls1 = Bidirectional(LSTM(256, return_sequences=True))(inp) + ls1 = Bidirectional(LSTM(128, return_sequences=True))(inp) ls2 = LSTM(128, return_sequences=True)(ls1) # ls3 = LSTM(32, return_sequences=True)(ls2) ls4 = LSTM(64)(ls2) # d1 = Dense(128, activation='relu')(ls4) - d2 = Dense(64, activation='relu')(ls4) + #d2 = Dense(64, activation='relu')(ls2) return Model(inp, ls4)