From f6c2d933c7c281379fd5bab5131a65856a54dec0 Mon Sep 17 00:00:00 2001 From: Sergio Guadarrama Date: Thu, 15 May 2014 18:02:08 -0700 Subject: [PATCH] Change ArgMaxLayerParam to ArgMaxParam for consitency --- src/caffe/test/test_argmax_layer.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/caffe/test/test_argmax_layer.cpp b/src/caffe/test/test_argmax_layer.cpp index f0754f0..627dd57 100644 --- a/src/caffe/test/test_argmax_layer.cpp +++ b/src/caffe/test/test_argmax_layer.cpp @@ -51,7 +51,7 @@ TYPED_TEST(ArgMaxLayerTest, TestSetup) { TYPED_TEST(ArgMaxLayerTest, TestSetupMaxVal) { LayerParameter layer_param; - ArgMaxLayerParameter* argmax_param = layer_param.mutable_argmax_param(); + ArgMaxParameter* argmax_param = layer_param.mutable_argmax_param(); argmax_param->set_out_max_val(true); ArgMaxLayer layer(layer_param); layer.SetUp(this->blob_bottom_vec_, &(this->blob_top_vec_)); @@ -86,7 +86,7 @@ TYPED_TEST(ArgMaxLayerTest, TestCPU) { TYPED_TEST(ArgMaxLayerTest, TestCPUMaxVal) { LayerParameter layer_param; Caffe::set_mode(Caffe::CPU); - ArgMaxLayerParameter* argmax_param = layer_param.mutable_argmax_param(); + ArgMaxParameter* argmax_param = layer_param.mutable_argmax_param(); argmax_param->set_out_max_val(true); ArgMaxLayer layer(layer_param); layer.SetUp(this->blob_bottom_vec_, &(this->blob_top_vec_)); -- 2.7.4