OrgLion-ML
/
dlib
mirror of https://github.com/davisking/dlib.git
1
0
Fork 0
Code Issues Releases Wiki Activity

fixed bug in tests

This commit is contained in:
Davis King 2016-04-25 07:01:36 -04:00
parent 24830e8e2c
commit 16583730d7
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
dlib/test/dnn.cpp
View File

@ -458,6 +458,7 @@ namespace
memcpy(A, truth); memcpy(A, truth);
DLIB_TEST(max(abs(mat(A)- mat(truth))) < 1e-5); DLIB_TEST(max(abs(mat(A)- mat(truth))) < 1e-5);
#ifdef DLIB_USE_CUDA
A = 4; A = 4;
A.device(); A.device();
B.host(); B.host();
@ -481,6 +482,7 @@ namespace
B.device(); B.device();
memcpy(A, truth); memcpy(A, truth);
DLIB_TEST(max(abs(mat(A)- mat(truth))) < 1e-5); DLIB_TEST(max(abs(mat(A)- mat(truth))) < 1e-5);
#endif
} }
{ {