Commit 962672b4 authored by Drozd Vadym's avatar Drozd Vadym
Browse files

fixed cpu mode

parent 41b559ac
...@@ -137,7 +137,7 @@ if __name__ == '__main__': ...@@ -137,7 +137,7 @@ if __name__ == '__main__':
caffe.set_mode_cpu() caffe.set_mode_cpu()
else: else:
caffe.set_mode_gpu() caffe.set_mode_gpu()
caffe.set_device(args.gpu_id) caffe.set_device(args.gpu_id)
net = caffe.Net(prototxt, caffemodel, caffe.TEST) net = caffe.Net(prototxt, caffemodel, caffe.TEST)
print '\n\nLoaded network {:s}'.format(caffemodel) 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