Merge pull request BVLC#6342 from Noiredd/gauss-fill-test-fix

Gaussian filler tests adjustments
tolgaeren · Apr 12, 2018 · 3066720 · 3066720
2 parents 106bfcf + 356a6cc
commit 3066720
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/caffe/test/test_filler.cpp b/src/caffe/test/test_filler.cpp
@@ -257,16 +257,16 @@ TYPED_TEST(GaussianFillerTest, TestFill) {
 }
 
 TYPED_TEST(GaussianFillerTest, TestFill1D) {
-  vector<int> blob_shape(1, 25);
-  const TypeParam tolerance = TypeParam(5);
+  vector<int> blob_shape(1, 125);
+  const TypeParam tolerance = TypeParam(3);
   this->test_params(blob_shape, tolerance);
 }
 
 TYPED_TEST(GaussianFillerTest, TestFill2D) {
   vector<int> blob_shape;
   blob_shape.push_back(8);
-  blob_shape.push_back(3);
-  const TypeParam tolerance = TypeParam(5);
+  blob_shape.push_back(15);
+  const TypeParam tolerance = TypeParam(3);
   this->test_params(blob_shape, tolerance);
 }