From a31b63cb7458ed07f1e655d86240e84bdfba11ff Mon Sep 17 00:00:00 2001 From: forresti Date: Wed, 15 Jan 2014 12:02:12 -0800 Subject: [PATCH] setting default power_wrapper batch size to match the imagenet_deploy prototxt --- python/caffe/imagenet/power_wrapper.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/caffe/imagenet/power_wrapper.py b/python/caffe/imagenet/power_wrapper.py index 825f398f749..5fa5c393bfc 100644 --- a/python/caffe/imagenet/power_wrapper.py +++ b/python/caffe/imagenet/power_wrapper.py @@ -32,7 +32,7 @@ CROP_MODES = ['center_only', 'corners', 'selective_search'] # NOTE: this must match the setting in the prototxt that is used! -BATCH_SIZE = 245 +BATCH_SIZE = 10 # Load the imagenet mean file IMAGENET_MEAN = np.load(