How to fit for different input size?

[mywebinfo65536]

Hi good job, could you tell me how to fit your code for input size, such as 128128， 192192， 256*256 ？ thanks.

[vztu]

Due to the nature of windowed attention, our model at 224x224 can only supports multipliers of 224 sizes, eg, 224/448/672/896... You'd have to manually pad the shape to satisfy the requirements. Please see our colab demos on how to inference at higher resolution.