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

fixed cpu mode

parent 41b559ac
......@@ -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