Solved by iacob in post #9 
              
              
                A network whose first layer is Conv2d will have an input size of (batchsize, n_channels, height, width). 
Since Convolutional layers in PyTorch are dynamic by design, there is no straightforward way to return the intended/expected height and width, and in fact (subject to remaining a valid size aft…