From c81a7b4468b13d0f51e1ecc85742b5e6d079feb2 Mon Sep 17 00:00:00 2001 From: Ubuntu Date: Fri, 17 Nov 2017 10:31:36 +0000 Subject: [PATCH] decreasing first layer node count to avoid gpu memory overflow --- speech_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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)