fix ArgMaxLayer bug in num bottom blobs decl. pointed out by @sguada

diff --git a/include/caffe/vision_layers.hpp b/include/caffe/vision_layers.hpp
index eb067bb..fc3dbbe 100644 (file)
--- a/include/caffe/vision_layers.hpp
+++ b/include/caffe/vision_layers.hpp
@@ -37,7 +37,7 @@ class ArgMaxLayer : public Layer<Dtype> {
   virtual inline LayerParameter_LayerType type() const {
     return LayerParameter_LayerType_ARGMAX;
   }
-  virtual inline int MinBottomBlobs() const { return 1; }
+  virtual inline int ExactNumBottomBlobs() const { return 1; }
   virtual inline int ExactNumTopBlobs() const { return 1; }
 
  protected: