From b2669d33437684e1861bf0ec07bbdb0f12e1fce0 Mon Sep 17 00:00:00 2001 From: Jeff Donahue Date: Sun, 8 Jun 2014 22:44:32 -0700 Subject: [PATCH] fix ArgMaxLayer bug in num bottom blobs decl. pointed out by @sguada --- include/caffe/vision_layers.hpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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: