Hi Marco,
My assignment is to first implement a LeNet-5 with 5x5 kernels.  I got that working.
Now I need to change to work with 3x3 kernels.  So I parameterized all kernels, except for pooling.  So, in that case, wouldn’t I want this line of code using the 3 as kernel_size?
x = x.view(-1, self.kernel_size * self.kernel_size * 16)