setting default power_wrapper batch size to match the imagenet_deploy…

… prototxt
BVLC · Jan 15, 2014 · aeca741 · aeca741
1 parent 3213bd8
commit aeca741
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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(