Модель VGG16 зависает компьютер

Question

В настоящее время я пытаюсь использовать модель vgg16 из библиотеки keras, но всякий раз, когда я создаю объект модели VGG16, выполняя

from keras.applications.vgg16 import VGG16
model = VGG16()

Я получаю следующее сообщение 3 раза.

tensorflow/core/framework/allocator.cc.124 allocation of 449576960 exceeds 10% of system memory

после этого мой компьютер зависает. Я использую 64-разрядную оперативную память объемом 4 ГБ с linux mint 18 и не имею доступа к графическому процессору.

Эта проблема связана с моей оперативной памятью?

В качестве временного решения я запускаю свои скрипты на python из командной строки, потому что мой компьютер там зависает меньше по сравнению с любой IDE. Кроме того, этого не происходит, когда я использую любую альтернативную модель, такую ​​как InceptionV3.

Я попробовал предоставленное решение здесь

но это не сработало

Любая помощь приветствуется.

Answer 1

Скорее всего, вам не хватает памяти (RAM). Попробуйте запустить top (или htop) параллельно и посмотрите, как используется ваша память.

В целом, модели VGG довольно большие и требуют приличного объема оперативной памяти. Тем не менее, фактическое требование зависит от размера партии. Меньшая партия означает меньший уровень активации.

Например, пакет из 6 изображений будет потреблять около 1 кг оперативной памяти ( ссылка ). В качестве теста вы можете уменьшить размер пакета до 1 и увидеть, что он умещается в вашей памяти.