Add padding to max pooling

2025-11-06 01:34:22 +00:00 · 2024-05-26 19:03:10 +02:00
parent 4a67b708f0
commit 94a16b4352
7 changed files with 78 additions and 19 deletions
--- a/test/model/test_model.cu
+++ b/test/model/test_model.cu
@@ -45,7 +45,7 @@ class ModelTest : public ::testing::Test {
        CUDANet::Layers::MaxPooling2d *maxpool2d =
            new CUDANet::Layers::MaxPooling2d(
                poolingInput, numFilters, poolingSize,
-                poolingStride, CUDANet::Layers::ActivationType::RELU
+                poolingStride, {0, 0}, CUDANet::Layers::ActivationType::RELU
            );
        model->addLayer("maxpool1", maxpool2d);