Почему при построении нейронной сети я получаю ошибку StopIteration

Question

введите описание изображения здесь Как я могу решить ошибку StopItate здесь. Я пытаюсь создать CNN для классификации изображений.

from keras.preprocessing.image import ImageDataGenerator
train_datagen = ImageDataGenerator(
    rescale=1./255,
    shear_range=0.2,
    zoom_range=0.2,
    horizontal_flip=True)
test_datagen = ImageDataGenerator(rescale=1./255)
train_set = train_datagen.flow_from_directory("training_set", target_size=(64, 64),
    batch_size=(64,64),
    class_mode='binary')
test_set = test_datagen.flow_from_directory( "test_set",
    target_size=(64, 64),
    batch_size=32,
    class_mode='binary')
model.fit_generator(
    train_set,
    steps_per_epoch=8000,
    epochs=25,
    validation_data=test_set,
    validation_steps=2000)