diff --git a/scripts/run-flownet-many.py b/scripts/run-flownet-many.py index 3c87fd4..768815d 100755 --- a/scripts/run-flownet-many.py +++ b/scripts/run-flownet-many.py @@ -74,11 +74,11 @@ def readTupleList(filename): tmp.flush() - if not args.verbose: - caffe.set_logging_disabled() - caffe.set_device(args.gpu) - caffe.set_mode_gpu() - net = caffe.Net(tmp.name, args.caffemodel, caffe.TEST) + if not args.verbose: + caffe.set_logging_disabled() + caffe.set_device(args.gpu) + caffe.set_mode_gpu() + net = caffe.Net(tmp.name, args.caffemodel, caffe.TEST) input_dict = {} for blob_idx in range(num_blobs):