Question

我有一个简单的循环，该循环在每次迭代中创建一个LSTM（具有相同的参数）并将其拟合到相同的数据。问题在于，在信号传递过程中会花费越来越多的时间。

batch_size = 10
optimizer = optimizers.adam(lr=0.001)
number_unites = 20

for counter_temp in range(1,100):
    t0 = time.time()

    model = None
    model = Sequential()
    model.add(LSTM(units=number_unites, batch_input_shape=(None, batch_size, 1), return_sequences=False))
    model.add(Dense(1))


    model.compile(loss='mean_absolute_error', optimizer=optimizer, metrics=['accuracy'])

    model.fit(x_train, y_train, epochs=50, validation_data=(x_train_val, y_train_val), verbose=0)

    print(time.time()-t0)

Answer 1

感谢DanielMöller，它已解决。我使用keras.backend.clear_session（）删除了“ Keras图”。

从keras.backend导入clear_session

clear_session（）

Keras拟合LSTM循环变慢

1 个答案: