Merge pull request ShaoqingRen#8 from drozdvadym/fix_cpu_mode

Fix CPU mode in Python demo
kukuruza · May 6, 2015 · b0758d0 · b0758d0
2 parents 41b559a + 962672b
commit b0758d0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/demo.py b/tools/demo.py
@@ -137,7 +137,7 @@ def parse_args():
         caffe.set_mode_cpu()
     else:
         caffe.set_mode_gpu()
-    caffe.set_device(args.gpu_id)
+        caffe.set_device(args.gpu_id)
     net = caffe.Net(prototxt, caffemodel, caffe.TEST)
 
     print '\n\nLoaded network {:s}'.format(caffemodel)