Does pytorch CNN care about image size?
I am playing with CNNs these days, and I have code like pasted below. My question is, would this work on any image size? It is not clear to me what parameter or channel, if any, cares about the image size? And if that’s the case, how does the model know how many neurons it needs, isn’t that a function of image size?