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

[Anonymous]

Я создаю нейронную сеть. Я не смог загрузить все данные обучения в память одновременно, поэтому я использую функцию tf.data.Dataset.from_generator TensorFlow для постепенной загрузки данных. Однако он выдает ошибку, сообщающую, что не принимает список тензоров в качестве типа.
TypeError: `output_signature` must contain objects that are subclass of
`tf.TypeSpec` but found which is not.

Входные данные для моей нейронной сети — это список из 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 must contain objects that are subclass of tf.TypeSpec but found which is not.


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