Keras错误:输入0与lstm_10层不兼容:预期ndim = 3,找到的ndim = 2

问题描述 投票:0回答:1

请和我一起对我很陌生。

我正在使用LSTM训练分类器,并具有以下代码

我在第三个LSTM层说尺寸存在问题时遇到问题

我的训练集的形状为(34799, 32, 32)

model = Sequential()
model.add(LSTM(64, activation = 'relu', input_shape=X_train[0].shape, return_sequences=True))
model.add(Dropout(0.25))
model.add(LSTM(128, activation = 'relu'))
model.add(Dropout(0.25))
model.add(LSTM(128, activation = 'relu'))
model.add(Dropout(0.25))
model.add(LSTM(64, activation = 'relu'))
model.add(Dropout(0.25))
model.add(Dense(len(sign_names), activation='softmax'))
python tensorflow keras lstm
1个回答
0
投票

每个LSTM层的返回序列都可以解决问题。

© www.soinside.com 2019 - 2024. All rights reserved.