setting default power_wrapper batch size to match the imagenet_deploy prototxt

forresti · forresti · commit aeca741a69d4 · 2014-01-15T12:02:12.000-08:00
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(
