From 886e4becc2fffb7ec5e8313e154a4da9ad6988cd Mon Sep 17 00:00:00 2001 From: liaogang Date: Mon, 8 May 2017 10:49:37 +0800 Subject: [PATCH] fix gpu_img_conv2 float precision issue --- paddle/gserver/tests/test_NetworkCompare.cpp | 3 +++ 1 file changed, 3 insertions(+) diff --git a/paddle/gserver/tests/test_NetworkCompare.cpp b/paddle/gserver/tests/test_NetworkCompare.cpp index 4db30f37a5b..ff1c195cee6 100644 --- a/paddle/gserver/tests/test_NetworkCompare.cpp +++ b/paddle/gserver/tests/test_NetworkCompare.cpp @@ -261,9 +261,12 @@ TEST(Compare, img_conv2) { std::string config_file_a = "./gserver/tests/img_conv_a.conf"; std::string config_file_b = "./gserver/tests/img_conv_c.conf"; bool useGpu = FLAGS_use_gpu; + double eps = FLAGS_checkgrad_eps; FLAGS_use_gpu = true; + FLAGS_checkgrad_eps = 1e-3; compareNetwork(config_file_a, config_file_b); FLAGS_use_gpu = useGpu; + FLAGS_checkgrad_eps = eps; } #endif -- GitLab