Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-06 09:44:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix inception v3 tests

Browse Source
This commit is contained in:
Matúš Námešný
2024-08-31 23:08:29 +02:00
parent bc9bff10cd
commit c8557fc0e4
9 changed files with 150 additions and 294 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
examples/inception_v3/tests/test_fixture.hpp
View File

@@ -6,6 +6,11 @@
#include "inception_v3.hpp"
bool __inline__ isCloseRelative(float a, float b, float rel_tol = 1e-3f, float abs_tol = 1e-3f) {
return std::abs(a - b) <= std::max(rel_tol * std::max(std::abs(a), std::abs(b)), abs_tol);
};
class InceptionBlockTest : public ::testing::Test {
protected:
CUDANet::Model *model;
@@ -42,7 +47,7 @@ class InceptionBlockTest : public ::testing::Test {
EXPECT_EQ(outputSize, expected.size());
for (int i = 0; i < outputSize; ++i) {
EXPECT_NEAR(expected[i], output[i], 1e-3f);
EXPECT_TRUE(isCloseRelative(expected[i], output[i]));
}
}
};