Set cfg.GPU_ID in demo.py

Author: rbgirshick

tools/demo.py

@@ -131,6 +131,7 @@ def parse_args():
     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)