Ошибка нехватки памяти при обучении модели с использованием Keras

Question

Я обучал модель, используя только тензор потока на моем GPU с помощью tf.device(), до сих пор она работала нормально. Я делаю другую модель NN в Керасе, и при использовании model.fit она выдает ошибку OOM.

Может кто-нибудь подсказать мне, как решить эту проблему?

Answer 1

Решено сейчас. Некоторое время я переключался на google.colab, но через некоторое время я использовал его на своем устройстве, и он отлично работал, я очистил некоторые файлы между ними и экспериментировал с моим графическим процессором, я думаю, что-то из этого вызвало решение. Все равно спасибо.

Answer 2

Ошибка OOM (Out of memory) возникает, когда вашей модели требуется больше памяти, чем доступно. Перед тренировкой проверьте память вашего GPU, занята ли она каким-либо процессом. В nvidia GPU вы можете проверить память с помощью команды nvidia-smi в cmd перед началом обучения.

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