Rework padding size setting

2025-11-06 01:34:22 +00:00 · 2024-04-20 16:31:28 +02:00
parent dbaab5652e
commit ecf7416f8e
5 changed files with 44 additions and 59 deletions
--- a/test/model/test_model.cu
+++ b/test/model/test_model.cu
@@ -24,10 +24,12 @@ class ModelTest : public ::testing::Test {
        CUDANet::Model *model =
            new CUDANet::Model(inputSize, inputChannels, outputSize);

+        int paddingSize = 0;
+
        // Conv2d
        CUDANet::Layers::Conv2d *conv2d = new CUDANet::Layers::Conv2d(
            inputSize, inputChannels, kernelSize, stride, numFilters,
-            CUDANet::Layers::Padding::VALID,
+            paddingSize,
            CUDANet::Layers::ActivationType::NONE
        );