From 74262309ad23c4306dc90ec6f4505e347b77c1c2 Mon Sep 17 00:00:00 2001
From: LordMathis <matus@namesny.com>
Date: Mon, 22 Apr 2024 20:58:02 +0200
Subject: [PATCH] Fix max pooling output size calc

---
 src/layers/max_pooling.cu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/layers/max_pooling.cu b/src/layers/max_pooling.cu
index d76ae77..6288637 100644
--- a/src/layers/max_pooling.cu
+++ b/src/layers/max_pooling.cu
@@ -15,7 +15,7 @@ MaxPooling2D::MaxPooling2D(
     : inputSize(inputSize), nChannels(nChannels), poolingSize(poolingSize), stride(stride) {
 
 
-    outputSize  = (inputSize - 1) / stride + 1;
+    outputSize  = (inputSize - poolingSize) / stride + 1;
 
     activation = new Activation(
         activationType, outputSize * outputSize * nChannels