I’m pretty new to pytorch, so I apologize if the question is very basic.
I have a model where, for each layer, I set the number of features, but the input image size is not fixed (it can change among trainings).
The last layer of my model is a 2D convolution that converts n input features to 1 value per pixel. To do this I would use a linear activation function.
The question is:
how to dynamically set the input/output number of features of a linear function considering that the input image can change among trainings and only the number of features is fixed?
Thanks a lot.