From 15ea096b89d997b0db48e73ab4351f0477abc14b Mon Sep 17 00:00:00 2001 From: Li Ding Date: Wed, 8 Nov 2017 14:53:00 -0500 Subject: [PATCH] fix run-flownet-many.py for reloading caffe model in every pass an improvement of speed for about 6x --- scripts/run-flownet-many.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) 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):