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

Set cfg.GPU_ID in demo.py

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