squeezeDet
squeezeDet copied to clipboard
How to run demo.py on CPU
So that's all the question ;)
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'):
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.