tryolabs / luminoth

Deep Learning toolkit for Computer Vision.

Home Page:https://tryolabs.com

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

First trial and OOM

jmFschneider opened this issue · comments

Hello,

I am just starting with with Luminoth and try "lumi predict image.jpg"

I am on a PC under unbutu with 8 giga octet of ram, a GPU with 2 giga and installed tensorflow-gpu.
The detection did not complet and I have the following message "ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[2000,14,14,1024] "

Withing Keras, I was able to modify the code to avoid this error. Is it to do some think here ?

Thnaks you for your help
Jean-Marie

You could try to reduce the post_nms_top_n in the config file which is by default 2000.