squeezeDet icon indicating copy to clipboard operation
squeezeDet copied to clipboard

How to run demo.py on CPU

Open mrlzla opened this issue 8 years ago • 2 comments

So that's all the question ;)

mrlzla avatar May 23 '17 07:05 mrlzla

I ran demo.py on CPU mode and GPU mode and I figured out that this model was faster on CPU rather GPU. What did I do wrong? I just changed line 21 in squeezeDet.py:

 with tf.device('/gpu:{}'.format(gpu_id)):

with:

 with tf.device('/cpu:0'):

mrlzla avatar May 31 '17 07:05 mrlzla

Could you provide some more information, like the specs of your machine, which tensorflow version you are using and maybe any errors or abnormalities you are seeing between running cpu vs gpu.

Timen avatar Jun 13 '17 11:06 Timen