From cbfa80d85f8638f5e1df19d513c8106d61ebacbc Mon Sep 17 00:00:00 2001 From: liuqi Date: Mon, 16 Apr 2018 14:14:30 +0800 Subject: [PATCH] Relax batchnorm and fc op test precision. --- mace/ops/batch_norm_test.cc | 4 ++-- mace/ops/fully_connected_test.cc | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/mace/ops/batch_norm_test.cc b/mace/ops/batch_norm_test.cc index d588eccd..cc122f7a 100644 --- a/mace/ops/batch_norm_test.cc +++ b/mace/ops/batch_norm_test.cc @@ -219,7 +219,7 @@ TEST_F(BatchNormOpTest, SimpleRandomHalfOPENCL) { ImageToBuffer(&net, "OutputImage", "OPENCLOutput", kernels::BufferType::IN_OUT_CHANNEL); - ExpectTensorNear(expected, *net.GetOutput("OPENCLOutput"), 1e-2, 1e-2); + ExpectTensorNear(expected, *net.GetOutput("OPENCLOutput"), 1e-1, 1e-2); } TEST_F(BatchNormOpTest, ComplexRandomOPENCL) { @@ -362,7 +362,7 @@ TEST_F(BatchNormOpTest, ComplexRandomHalfOPENCL) { ImageToBuffer(&net, "OutputImage", "OPENCLOutput", kernels::BufferType::IN_OUT_CHANNEL); - ExpectTensorNear(expected, *net.GetOutput("OPENCLOutput"), 1e-2, 1e-2); + ExpectTensorNear(expected, *net.GetOutput("OPENCLOutput"), 1e-1, 1e-2); } TEST_F(BatchNormOpTest, NEONTest) { diff --git a/mace/ops/fully_connected_test.cc b/mace/ops/fully_connected_test.cc index d4c17fb4..33c83927 100644 --- a/mace/ops/fully_connected_test.cc +++ b/mace/ops/fully_connected_test.cc @@ -253,7 +253,7 @@ void TestWXFormat(const index_t batch, 1e-1, 1e-1); } else { ExpectTensorNear(expected, *net.GetOutput("OPENCLOutput"), - 1e-3, 1e-3); + 1e-2, 1e-3); } } } // namespace -- GitLab