Commit b0758d0a authored by Ross Girshick's avatar Ross Girshick

Merge pull request #8 from drozdvadym/fix_cpu_mode

Fix CPU mode in Python demo
parents 41b559ac 962672b4
......@@ -137,7 +137,7 @@ if __name__ == '__main__':
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)
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment