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