Add epsilon param to batch norm

2025-11-06 01:34:22 +00:00 · 2024-05-19 15:13:22 +02:00
parent c84f58b97c
commit 4a1c4a5f91
3 changed files with 3 additions and 3 deletions
--- a/test/layers/test_batch_norm.cu
+++ b/test/layers/test_batch_norm.cu
@@ -13,7 +13,7 @@ TEST(BatchNormLayerTest, BatchNormSmallForwardTest) {
    cudaError_t cudaStatus;

    CUDANet::Layers::BatchNorm batchNorm(
-        inputSize, nChannels, CUDANet::Layers::ActivationType::NONE
+        inputSize, nChannels, 1e-5f, CUDANet::Layers::ActivationType::NONE
    );

    std::vector<float> weights = {0.63508f, 0.64903f};