diff --git a/include/caffe/vision_layers.hpp b/include/caffe/vision_layers.hpp index eb067bb624e..fc3dbbe1938 100644 --- a/include/caffe/vision_layers.hpp +++ b/include/caffe/vision_layers.hpp @@ -37,7 +37,7 @@ class ArgMaxLayer : public Layer { 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: