ImageDataGenerator в Google Colab работает вечно

Question

Я запускаю простую проблему классификации изображений в Google Colab с Keras. Чтобы увеличить размер моих тренировочных данных, я использую ImageDataGenerator.

Когда я запускаю следующий код ниже на моем локальном компьютере, он работает без проблем. Тем не менее, когда я запускаю его на Google Colab, он, кажется, работает бесконечно.

augmented_history = model.fit_generator(datagen.flow(x_train, 
                                                     y_train_wide, 
                                                     batch_size=32),
                                        epochs=15,
                                        validation_data=(x_valid, y_valid_wide),
                                        callbacks=[es, mc])

На Google Colab, он выводит на консоль следующее.

Epoch 1/15
   30514/Unknown - 85s 44ms/step - loss: 0.0245 - accuracy: 0.9834

Кажется, никогда не остановиться.