Python Записные книжки Google Colab, файл notfounderror при использовании ImageDataGenerator Flow_from_directory

Question

При попытке запустить ImageDataGenerator flow_from_directory в Google Colab Notebooks я получаю сообщение об ошибке "файл не найден".

Странно то, что он работает при первом запуске, но затем, когда я перезагружаю данные в base_dir, он выдает ошибка.

Вот код, который я использую для чтения данных в блокнотах

_URL = "http://xxxxx/images/Model_images_40.tar.gz"

zip_file = tf.keras.utils.get_file(origin=_URL, 
                                   fname="Model_images_40.tar.gz", 
                                   extract=True)

base_dir = os.path.join(os.path.dirname(zip_file), 'Model_images_40')

Вот как выглядит base_dir

'/root/.keras/datasets/Model_images_40'

Тогда вот следующий бит кода, который я запускаю, чтобы получить данные об обучении и тестировании (но он дает ошибку ниже)

IMAGE_SIZE = 224
BATCH_SIZE = 64

datagen = tf.keras.preprocessing.image.ImageDataGenerator(
    rescale=1./255, 
    validation_split=0.35)

train_generator = datagen.flow_from_directory(
    base_dir,
    target_size=(IMAGE_SIZE, IMAGE_SIZE),
    batch_size=BATCH_SIZE,
    subset='training' 
    )

val_generator = datagen.flow_from_directory(
    base_dir,
    target_size=(IMAGE_SIZE, IMAGE_SIZE),
    batch_size=BATCH_SIZE, 
    subset='validation')

Вот ошибка

FileNotFoundError                         Traceback (most recent call last)
<ipython-input-24-c8c878b374f7> in <module>()
     10     target_size=(IMAGE_SIZE, IMAGE_SIZE),
     11     batch_size=BATCH_SIZE,
---> 12     subset='training'
     13     )
     14 

1 frames
/usr/local/lib/python3.6/dist-packages/keras_preprocessing/image/directory_iterator.py in __init__(self, directory, image_data_generator, target_size, color_mode, classes, class_mode, batch_size, shuffle, seed, data_format, save_to_dir, save_prefix, save_format, follow_links, subset, interpolation, dtype)
    104         if not classes:
    105             classes = []
--> 106             for subdir in sorted(os.listdir(directory)):
    107                 if os.path.isdir(os.path.join(directory, subdir)):
    108                     classes.append(subdir)

FileNotFoundError: [Errno 2] No such file or directory: '/root/.keras/datasets/Model_images_40'

Я полагаю, что это относительно тривиально, поскольку я Я новичок в python и ноутбуках Colab и не совсем понимаю этот код.