fixed cpu mode

danilons · May 6, 2015 · 962672b · 962672b
1 parent 41b559a
commit 962672b
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)