TensorFlow не принимает тип списка для генератора набора данных ⇐ Python

[Anonymous]

Я создаю нейронную сеть. Я не мог загрузить все данные обучения в память одновременно, поэтому я использую функцию tf.data.Dataset.from_generator TensorFlow для постепенной загрузки данных. Однако он выдает ошибку, сообщающую, что не принимает список тензоров в качестве типа.
TypeError: output_signature должен содержать объекты, которые являются подклассом tf.TypeSpec > но нашел , а это не так.
Входные данные для моей нейронной сети представляют собой список из 151 отдельного тензора. Как я могу представить это в генераторе? Мой код ниже:

Код: Выделить всё

def generator(file_paths, batch_size, files_per_batch, tam, value):
return tf.data.Dataset.from_generator(
lambda: data_generator(file_paths, batch_size, files_per_batch, tam, value),
output_signature=(
[tf.TensorSpec(shape=(batch_size, tam), dtype=tf.float32) for _ in range(tam+1)],  # Lista de 151 tensores
tf.TensorSpec(shape=(batch_size, tam), dtype=tf.float32)  # Rótulos
)
)

inputArray = [Input(shape=(tam,)) for _ in range(tam + 1)]

train_dataset = generator(file_paths, batch_size, files_per_batch, tam, False)
train_dataset = train_dataset.prefetch(tf.data.AUTOTUNE)

model.fit(train_dataset, epochs=1000, validation_split=0.2, verbose=1)

Я пытался использовать tf.data.Dataset.from_generator для пакетной загрузки данных в мою нейронную сеть, поскольку я не могу загрузить все данные в память одновременно.
Однако я столкнулся с ошибкой: TypeError: output_signature должен содержать объекты, которые являются подклассом tf.TypeSpec, но найден , который не является таковым.


Подробнее здесь: https://stackoverflow.com/questions/792 ... -generator