Использование оперативной памяти TensorFlow ⇐ Python

[Anonymous]

Я делаю проект искусственного интеллекта на TensorFlow, код скопировал из сети. Проблема в том, что когда я пытаюсь подогнать модель, она использует всю предоставленную оперативную память (12,7 ГБ). Я запускаю ее в Google Colab.
Мой код:

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

history=model.fit(
train_ds,
epochs=EPOCHS,
batch_size=BATCH_SIZE,
verbose=1,
validation_data=val_ds
)

train_ds — это список, содержащий 20665 изображений.
epochs=50
batch_size=32
val_ds — это список, содержащий 2583 изображения.
Что мне следует сделать, чтобы решить эту проблему? Он заполняет оперативную память в течение первых 30 секунд или около того и даже не начинает выполнение. Также изображения предварительно обрабатываются до размера 256x256

Подробнее здесь: https://stackoverflow.com/questions/791 ... tensorflow