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(