I am having the same problem coming from TF/Keras: for pytorch conv models i can`t tell what the input size should be when i use one implementation or another.
Relying on google searches seems awkward since the input tensor size should be a property of the network/architecture. I am worried might have missed something about using with torch.
Isn`t there any built in methods to query the input size or at least to help deduce it?
I even had a problem with the qubvel implementation of Unet w/ effnet encoder where the effnet-b2 encoder got shape mismatches for 260x260 imgs (the official size) but worked for 256.