bourdakos1 / Custom-Object-Detection

Custom Object Detection with TensorFlow

Home Page:https://medium.freecodecamp.org/tracking-the-millenium-falcon-with-tensorflow-c8c86419225e

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Consumes whole ram and freeze system

gaurav10069 opened this issue · comments

Hi you have given a very good explanation. But I have some Issue in training.
I have followed your instruction but When I run train.py with the dataset given by you my system freezes at step 0 and then I have to restart my system.
When I checked ram usage using htop command, I saw It consumes whole RAM.
I am using linux 14.04 with 8 gb, 4 CPU cores.
Can we run this code on CPU or it is only compatible to GPU?

You should be able to run it on either CPU or GPU, but GPU will run faster. You can lower the batch size if your don't have enough RAM.