Remove bug

Paupiera · Jan 9, 2024 · facb3bc · facb3bc
1 parent 704b6c0
commit facb3bc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cnn_mnist/models/model.py b/cnn_mnist/models/model.py
@@ -18,7 +18,7 @@ def __init__(self, number_of_kernels, kernel_sizes, input_channels, xsize_input)
             self.convlayers.append(nn.Conv2d(input_channels, n_kernels, kernel_size))
 
             input_channels = n_kernels
-            xsize_input -= 1  # 2
+            xsize_input -= 2
 
         self.last_feature_xdim = xsize_input
         self.relu = nn.ReLU()