Keras зависает при создании моделей через цикл for

Question

Я в настоящее время путаюсь со следующими фрагментами кода: все фрагменты являются частью учебника отсюда .

У меня есть функция, которая компилирует и подгоняет новую модель keras, используя Sequential API:

def fit_model(trainX, trainy):
    trainy_enc = to_categorical(trainy)
    # define model
    model = Sequential()
    model.add(Dense(25, input_dim=2, activation='relu'))
    model.add(Dense(3, activation='softmax'))
    model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy'])
    # fit model
    model.fit(trainX, trainy_enc,epochs=200, verbose=0)
    return model

Теперь я хочу создать, например, 10 моделей, используя кратчайший путь для создания, компиляции и подгонки моделей и сохранения их в списке:

n_members = 10
members = [fit_model(trainX, trainy) for _ in range(n_members)]

Проблема в том, что Керас (случайно), кажется, зависает, и прогресс в обучении больше не продолжается. Это действительно случайно, иногда это происходит на первой модели, иногда на пятой. И эпохи, в которых он висит, тоже произвольны.

Как мне выйти из этой проблемы? Я не уверен, что все реализовано, как в исходном учебнике, и там, похоже, так и работает.