That’s strange. I’m testing it with images of size [3, 690, 334] and it seems to work:
Using 34 337 256 256
Using 34 337 256 256
Using 72 102 256 256
Using 72 102 256 256
Using 57 382 256 256
Using 57 382 256 256