add split layer tests
authorJeff Donahue <jeff.donahue@gmail.com>
Sat, 15 Feb 2014 20:24:00 +0000 (12:24 -0800)
committerJeff Donahue <jeff.donahue@gmail.com>
Sat, 15 Feb 2014 23:05:23 +0000 (15:05 -0800)
src/caffe/test/test_split_layer.cpp

index 894f4c5..9a3a2d3 100644 (file)
@@ -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_);
 }
 
 }