From 37c29d673493291a026b6b816d5629fca7768183 Mon Sep 17 00:00:00 2001
From: LordMathis <matus@namesny.com>
Date: Thu, 23 May 2024 21:38:11 +0200
Subject: [PATCH] Fix wrong conv2d output size calculation

---
 src/layers/conv2d.cu | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/layers/conv2d.cu b/src/layers/conv2d.cu
index bbcc516..b7e83a3 100644
--- a/src/layers/conv2d.cu
+++ b/src/layers/conv2d.cu
@@ -26,11 +26,12 @@ Conv2d::Conv2d(
       stride(stride),
       numFilters(numFilters),
       paddingSize(paddingSize) {
+
     outputSize = {
         (inputSize.first - kernelSize.first + 2 * paddingSize.first) /
                 stride.first + 1,
-        (inputSize.first - kernelSize.first + 2 * paddingSize.first) /
-                stride.first + 1
+        (inputSize.second - kernelSize.second + 2 * paddingSize.second) /
+                stride.second + 1
     };
 
     activation =