Input size is not a square, what should I do in this line?

[JerryKingQAQ]

Thanks to your excellent work!
There has a tensor which size is (3, 56, 112), What should I do in this line to modify ?
self.num_patches = (img_size // (4 * 2 * 2)) * (img_size // (4 * 2 * 2)) # there are 3 sfot split, stride are 4,2,2 seperately

[holderhe]

same issue here. after t2t, the num_patches is not same as initialization. how to deal with it?