Ошибка компиляции https://www.tensorflow.org/tutorials/images/classification

Question

Я скопировал код из этого урока в проект pycharm и настроил свой venv с помощью tenorflow и matplotlib. Когда я пытаюсь скомпилировать код, я получаю следующую ошибку:

    Traceback (most recent call last):
    File "<input>", line 1, in <module>
    File "C:\Program Files\JetBrains\PyCharm 2019.3.2\plugins\python\helpers\pydev\_pydev_bundle\pydev_umd.py", line 197, in runfile
pydev_imports.execfile(filename, global_vars, local_vars)  # execute the script
    File "C:\Program Files\JetBrains\PyCharm 2019.3.2\plugins\python\helpers\pydev\_pydev_imps\_pydev_execfile.py", line 18, in execfile
    exec(compile(contents+"\n", file, 'exec'), glob, loc)
    File "C:/Users/sirch/PycharmProjects/Cats_vs_Dogs/venv/Test_NN.py", line 52, in <module>
sample_training_images, _ = next(train_data_gen)
    File "C:\Users\sirch\PycharmProjects\Cats_vs_Dogs\venv\lib\site-packages\keras_preprocessing\image\iterator.py", line 104, in __next__
return self.next(*args, **kwargs)
    File "C:\Users\sirch\PycharmProjects\Cats_vs_Dogs\venv\lib\site-packages\keras_preprocessing\image\iterator.py", line 116, in next

Из сообщений об ошибках я предполагаю, что проблема связана со следующей строкой кода:

    train_image_generator = ImageDataGenerator(rescale=1./255) # Generator for our training data
    validation_image_generator = ImageDataGenerator(rescale=1./255) # Generator for our validation data
    train_data_gen = train_image_generator.flow_from_directory(batch_size=batch_size,
                                                       directory=train_dir,
                                                       shuffle=True,
                                                       target_size=(IMG_HEIGHT, IMG_WIDTH),
                                                       class_mode='binary')
    val_data_gen = validation_image_generator.flow_from_directory(batch_size=batch_size,
                                                          directory=validation_dir,
                                                          target_size=(IMG_HEIGHT, IMG_WIDTH),
                                                          class_mode='binary')
    sample_training_images, _ = next(train_data_gen)

Любая идея что это за ошибка на самом деле или почему она происходит?