malar
/
speech-scoring
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

decreasing first layer node count to avoid gpu memory overflow

master
Ubuntu 2017-11-17 10:31:36 +00:00
parent c682962c8f
commit c81a7b4468
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
speech_model.py
View File

@ -17,12 +17,12 @@ def create_base_rnn_network(input_dim):
''' '''
inp = Input(shape=input_dim) inp = Input(shape=input_dim)
# ls0 = LSTM(512, return_sequences=True)(inp) # 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) ls2 = LSTM(128, return_sequences=True)(ls1)
# ls3 = LSTM(32, return_sequences=True)(ls2) # ls3 = LSTM(32, return_sequences=True)(ls2)
ls4 = LSTM(64)(ls2) ls4 = LSTM(64)(ls2)
# d1 = Dense(128, activation='relu')(ls4) # d1 = Dense(128, activation='relu')(ls4)
d2 = Dense(64, activation='relu')(ls4) #d2 = Dense(64, activation='relu')(ls2)
return Model(inp, ls4) return Model(inp, ls4)