Как сохранять и использовать CIFAR10 с локального диска?

Пробую сделать классификатор изображений по обучающим статьям в сети - на основе Tensorflow и датасета CIFAR10.

В коде есть такие строки, загружающие датасет:

import tensorflow_datasets as tfds
ds_train, info = tfds.load("cifar10", with_info=True, split="train", as_supervised=True)
ds_test = tfds.load("cifar10", split="test", as_supervised=True)

Проблема в том, что при запуске программа начинает долго загружать и распаковывать файлы и в процессе соединение прерывается (возможно сказываются какие-либо блокировки провайдеров и тд).

Я нашел сайт авторов датасета и скачал его с сайта. Получились такие файлы:

batches.meta - 158 кб

data_batch_1 - 31 мб

data_batch_2 - 31 мб

data_batch_3 - 31 мб

data_batch_4 - 31 мб

data_batch_5 - 31 мб

test_batch - 31 мб

Подскажите - как именно нужно переписать эту часть кода, чтобы можно было загружать датасет сразу из папки на локальном диске из вот этих файлов?