Keras, подходящие LSTM, становятся медленнее в цикле - PullRequest
Новая
       33

Keras, подходящие LSTM, становятся медленнее в цикле

0 голосов
/ 12 декабря 2018

У меня есть простой цикл, который на каждой итерации создает 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)

1 Ответ

0 голосов
/ 12 декабря 2018

Благодаря DanielMöller, это решено.Я использовал keras.backend.clear_session () для удаления «графика Кераса».

из keras.backend import clear_session

clear_session ()

...