From 4bed0ac9bab45246183184d84ff1b742e60574c7 Mon Sep 17 00:00:00 2001 From: Jeff Donahue Date: Mon, 24 Aug 2015 19:44:18 -0700 Subject: [PATCH] TestConcatLayer: add gradient check for bottom[1] only (to verify propagate_down[0] == false correctness) --- src/caffe/test/test_concat_layer.cpp | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/src/caffe/test/test_concat_layer.cpp b/src/caffe/test/test_concat_layer.cpp index 662a50f..088e0a4 100644 --- a/src/caffe/test/test_concat_layer.cpp +++ b/src/caffe/test/test_concat_layer.cpp @@ -173,4 +173,13 @@ TYPED_TEST(ConcatLayerTest, TestGradientChannels) { this->blob_top_vec_); } +TYPED_TEST(ConcatLayerTest, TestGradientChannelsBottomOneOnly) { + typedef typename TypeParam::Dtype Dtype; + LayerParameter layer_param; + ConcatLayer layer(layer_param); + GradientChecker checker(1e-2, 1e-2); + checker.CheckGradient(&layer, this->blob_bottom_vec_0_, + this->blob_top_vec_, 1); +} + } // namespace caffe -- 2.7.4