CODEWITHSUNDEEP
python-3.xkerasdeep-learningvgg-nettransfer-learning
Aditya rawat
Aditya rawat

Reputation: 61

VGG16 model freezes computer

I am currently trying to use the vgg16 model from keras library but whenever I create an object of the VGG16 model by doing

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

I get the following message 3 times.

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

following this, my computer freezes. I am using a 64-bit, 4gb RAM with linux mint 18 and I have no access to GPU.

Is this problem has to do something with my RAM?

As a temporary solution I am running my python scripts from command line because my computer freezes less there compared to any IDE. Also, this does not happen when I use any alternate model like InceptionV3.

I have tried the solution provided here

but it didn't work

Any help is appreciated.

Upvotes: 2

Views: 459

Answers (1)

Tautvydas
Tautvydas

Reputation: 2067

You are most likely running out of memory (RAM). Try running top (or htop) in parallel and see your memory utilization.

In general, VGG models are rather big and require a decent amount of RAM. That said, the actual requirement depends on batch size. Smaller batch means smaller activation layer.

For example, a 6 image batch would consume about a gig of ram (reference). As a test you could lower your batch size to 1 and see it that fits in your memory.

Upvotes: 2

Related Questions