From eb56c5b65257d0020d559be3688508064a9d018b Mon Sep 17 00:00:00 2001 From: sguada Date: Wed, 26 Feb 2014 22:33:57 -0800 Subject: [PATCH] Fixed test to pass and don't exhaustive search since it takes too long --- src/caffe/test/test_concat_layer.cpp | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/caffe/test/test_concat_layer.cpp b/src/caffe/test/test_concat_layer.cpp index 472c936..aeee81d 100644 --- a/src/caffe/test/test_concat_layer.cpp +++ b/src/caffe/test/test_concat_layer.cpp @@ -110,7 +110,8 @@ TYPED_TEST(ConcatLayerTest, TestCPUGradient) { GradientChecker checker(1e-2, 1e-3); // it is too expensive to call curand multiple times, so we don't do an // exhaustive gradient check. - checker.CheckGradient(layer, this->blob_bottom_vec_0, this->blob_top_vec_); + checker.CheckGradient(&layer, &(this->blob_bottom_vec_0), + &(this->blob_top_vec_)); } TYPED_TEST(ConcatLayerTest, TestGPUGradient) { @@ -120,7 +121,8 @@ TYPED_TEST(ConcatLayerTest, TestGPUGradient) { GradientChecker checker(1e-2, 1e-3); // it is too expensive to call curand multiple times, so we don't do an // exhaustive gradient check. - checker.CheckGradient(layer, this->blob_bottom_vec_0, this->blob_top_vec_); + checker.CheckGradient(&layer, &(this->blob_bottom_vec_0), + &(this->blob_top_vec_)); } -- 2.7.4