add split layer tests

winstywang · Feb 26, 2014 · 5ca5ef5 · 5ca5ef5
1 parent f0562d7
commit 5ca5ef5
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/caffe/test/test_split_layer.cpp b/src/caffe/test/test_split_layer.cpp
@@ -92,6 +92,8 @@ TYPED_TEST(SplitLayerTest, TestCPUGradient) {
   SplitLayer<TypeParam> layer(layer_param);
   GradientChecker<TypeParam> checker(1e-2, 1e-2);
   checker.CheckGradientExhaustive(layer, this->blob_bottom_vec_, this->blob_top_vec_);
+  checker.CheckGradientExhaustive(layer, this->blob_bottom_vec_,
+      this->blob_top_vec_);
 }
 
 TYPED_TEST(SplitLayerTest, TestGPUGradient) {
@@ -100,6 +102,8 @@ TYPED_TEST(SplitLayerTest, TestGPUGradient) {
   SplitLayer<TypeParam> layer(layer_param);
   GradientChecker<TypeParam> checker(1e-2, 1e-2);
   checker.CheckGradientExhaustive(layer, this->blob_bottom_vec_, this->blob_top_vec_);
+  checker.CheckGradientExhaustive(layer, this->blob_bottom_vec_,
+      this->blob_top_vec_);
 }
 
 }