diff options
| author | Jeff Donahue <jeff.donahue@gmail.com> | 2014-02-15 12:24:00 -0800 |
|---|---|---|
| committer | Jeff Donahue <jeff.donahue@gmail.com> | 2014-02-15 15:05:23 -0800 |
| commit | d86f9b0ce94b45d899a077ebeb47e78e257fa7df (patch) | |
| tree | 2cc3e1f2501d1d6ee902ce658e818b9929f3e226 /src | |
| parent | 3231853b9edee7bf299b06d95b0caa0548e2f506 (diff) | |
| download | caffeonacl-d86f9b0ce94b45d899a077ebeb47e78e257fa7df.tar.gz caffeonacl-d86f9b0ce94b45d899a077ebeb47e78e257fa7df.tar.bz2 caffeonacl-d86f9b0ce94b45d899a077ebeb47e78e257fa7df.zip | |
add split layer tests
Diffstat (limited to 'src')
| -rw-r--r-- | src/caffe/test/test_split_layer.cpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/caffe/test/test_split_layer.cpp b/src/caffe/test/test_split_layer.cpp index 894f4c50..9a3a2d37 100644 --- 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_); } } |