Skip to content

Commit b0758d0

Browse files
committed
Merge pull request rbgirshick#8 from drozdvadym/fix_cpu_mode
Fix CPU mode in Python demo
2 parents 41b559a + 962672b commit b0758d0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tools/demo.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -137,7 +137,7 @@ def parse_args():
137137
caffe.set_mode_cpu()
138138
else:
139139
caffe.set_mode_gpu()
140-
caffe.set_device(args.gpu_id)
140+
caffe.set_device(args.gpu_id)
141141
net = caffe.Net(prototxt, caffemodel, caffe.TEST)
142142

143143
print '\n\nLoaded network {:s}'.format(caffemodel)

0 commit comments

Comments
 (0)