m53297601
/
Paddle
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge pull request #2068 from gangliao/gpu_conv

Browse Source 
Adjust float precision in img_conv2 for unit test
feature/design_of_v2_layer_converter
gangliao 8 years ago committed by GitHub
parent f003a63bfd 6dc58cdd25
commit aaef5867dd
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File

3
paddle/gserver/tests/test_NetworkCompare.cpp
Unescape View File

@ -261,9 +261,12 @@ TEST(Compare, img_conv2) {
std::string config_file_a = "./gserver/tests/img_conv_cudnn.py"; std::string config_file_a = "./gserver/tests/img_conv_cudnn.py";
std::string config_file_b = "./gserver/tests/img_conv_exconv.py"; std::string config_file_b = "./gserver/tests/img_conv_exconv.py";
bool useGpu = FLAGS_use_gpu; bool useGpu = FLAGS_use_gpu;
double eps = FLAGS_checkgrad_eps;
FLAGS_use_gpu = true; FLAGS_use_gpu = true;
FLAGS_checkgrad_eps = 1e-2;
compareNetwork(config_file_a, config_file_b); compareNetwork(config_file_a, config_file_b);
FLAGS_use_gpu = useGpu; FLAGS_use_gpu = useGpu;
FLAGS_checkgrad_eps = eps;
} }
#endif #endif

Write Preview
Loading…
Cancel
Save