search

bourdakos1 / Custom-Object-Detection

Custom Object Detection with TensorFlow
https://medium.freecodecamp.org/tracking-the-millenium-falcon-with-tensorflow-c8c86419225e
MIT License
347 stars 181 forks source link

Consumes whole ram and freeze system #15

Closed gaurav10069 closed 6 years ago

gaurav10069 commented 6 years ago

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?

bourdakos1 commented 6 years ago

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.