Fix batch norm layer

2025-11-06 01:34:22 +00:00 · 2024-05-14 21:42:01 +02:00
parent da8f3167cb
commit 33a3c6afcb
2 changed files with 4 additions and 12 deletions
--- a/test/layers/test_batch_norm.cu
+++ b/test/layers/test_batch_norm.cu
@@ -71,7 +71,7 @@ TEST(BatchNormLayerTest, BatchNormSmallForwardTest) {

    // std::cout << "BatchNorm: " << std::endl;
    for (int i = 0; i < output.size(); i++) {
-        EXPECT_EQ(output[i], expected[i]);
+        EXPECT_NEAR(output[i], expected[i], 1e-5);
        // std::cout << output[i] << " ";
    }
    // std::cout << std::endl;