Merge pull request BVLC#34 from forresti/master

set default power_wrapper batch size to match the imagenet_deploy compatible out-of-the-box with the ipython notebook tutorial
mitmul · Jan 19, 2014 · 465e9cb · 465e9cb
2 parents 20625b8 + a31b63c
commit 465e9cb
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(