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

mitmul · Jun 9, 2014 · b2669d3 · b2669d3
1 parent d292b38
commit b2669d3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: