neural-network - Keras LSTM 让我的电脑崩溃了？

Question

我正在尝试学习 Keras，并制作了一个看起来非常简单的模型，只是为了看看我会遇到什么样的错误。

input_layer = Input(shape=inp_size)

dens_layer = Dense(10000)(input_layer)
dens_layer_2 = Dense(10000)(dens_layer)
lstm_1 = LSTM(10000)(dens_layer_2)
lstm_2 = LSTM(10000)(lstm_1)
dense_layer = Dense(10000)(lstm_1)
dense_layer_2 = Dense(10000)(dense_layer)
output_layer = Dense(2)(dense_layer_2)

Dens_layer 是在 2 秒内构建的，而 dens_layer_2 是在 0.07 秒内构建的，但是当我初始化第一个 LSTM 层时，它只是继续做……某事……直到我的计算机突然关闭并重新启动。它使我的计算机变慢了一点，另一个答案建议是操作系统交换，但我不明白为什么我的计算机会突然重新启动。

标签： neural-networkkeraslstm

Answer 1

10000 个单位确实很多，它可能需要比你拥有的更多的资源。相比之下，ImageNet CNN 中的大多数 Dense 层都有 4096 个单元。