Hi,
For my model my input (image) needs to be divisible by 32 and I would like to pad my input dynamically to fit this requirement. Meaning if the input would be for example 520x520x3, I want it to be padded to 544x544x3.
torch.nn.functional.pad() requires the pad to be list of ints, but my paddings would be a tensor computed dynamically depending on the shape of the input.
The model (including the padding) is supposed to be exported to onnx so it has to be traceable.
Does anybody have a clue on how to approach this problem?
Thanks in advance,
Andreas