Is there any requirements on the size of input images

[Tord-Zhang]

When I input two images of size 768 x 1856 to the model, I got the error below:
einops.EinopsError: Error while processing rearrange-reduction pattern "(y v) d -> y () v d". Input tensor shape: torch.Size([25600, 128]). Additional info: {'y': 232}. Shape mismatch, can't divide axis of length 25600 in chunks of 232
Any idea why this happens?

[zacjiang]

I have tested with the demo.sh script with two 768 x 1856 images and it didn't seem to give an error.

[AlphonsG]

@Tord-Zhang If you're not using the demo script, make sure the images are padded like here.