In a nn.module, for a model which fuses feature maps of different strides, I am wondering if it is possible to do nn.Upsample, not with the scale parameter, but with the size parameter.
This size parameter would be dependent on which level of feature map is being upsampled, as well as the input image size which you only see during the forward pass and not when the module is instantiated.
The reason not to use scale is that I would like to be able to work with images of any sizes, without having to resize to some powers of 2.