Hi,
thanks for replying.
I use the same code to train different models over different set of images having different shapes (among datasets, of course).
So, if the final 2D convolution has 32 features as output, and the image has shape 512x512, I have to initialize the linear module to 32x512x512 input features and 1x512x512 a output features?
Do I need also a “view” to reshape the output to 512x512 shape?
What if I do not set any activation function for a 2D convolution?
Which is the default one?
Thanks a lot.
Paolo