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

Cleanup and refactor

Browse Source
This commit is contained in:
Matúš Námešný
2024-04-11 22:52:41 +02:00
parent 4b9d123e94
commit 18522c2dea
9 changed files with 81 additions and 72 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/model/test_model.cu
View File

@@ -105,11 +105,9 @@ TEST(Model, TestModelPredict) {
// float sum = 0.0f;
for (int i = 0; i < outputSize; ++i) {
sum += output[i];
std::cout << output[i] << " ";
}
std::cout << std::endl;
EXPECT_NEAR(sum, 1.0f, 1e-2f);
EXPECT_NEAR(sum, 1.0f, 1e-5f);
cudaDeviceReset();
}