Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-05 17:34:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix max pooling output size calc

Browse Source
This commit is contained in:
Matúš Námešný
2024-04-22 20:58:02 +02:00
parent c2acad151b
commit 74262309ad
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/layers/max_pooling.cu
View File

@@ -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