Commit 81ee2272 authored by Ross Girshick's avatar Ross Girshick
Browse files

Set cfg.GPU_ID in demo.py

parent 19062d51
...@@ -131,6 +131,7 @@ if __name__ == '__main__': ...@@ -131,6 +131,7 @@ if __name__ == '__main__':
else: else:
caffe.set_mode_gpu() caffe.set_mode_gpu()
caffe.set_device(args.gpu_id) caffe.set_device(args.gpu_id)
cfg.GPU_ID = 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