2017-07-23 80 views
0

如何爲以下神經網絡添加另一個LSTM圖層?在tflearn中添加多個LSTM圖層

net = tflearn.input_data([None, width, height]) 
net = tflearn.lstm(net, 128) 
net = tflearn.fully_connected(net, classes, activation='softmax') 
net = tflearn.regression(net, optimizer='adam', learning_rate=learning_rate, loss='categorical_crossentropy') 
model = tflearn.DNN(net, tensorboard_verbose=3,tensorboard_dir='log') 

`

+0

加上'淨= tflearn.lstm(淨值,128)之間的' lstm和完全連接的圖層 –

回答

0
net = tflearn.lstm(net, 128, return_seq=True) 
net = tflearn.lstm(net, 256) 

不要忘記設置return_seq=True以前的LSTM層上,或者你會得到一個錯誤

相關問題