CCT and non-square images
Gasp34 opened this issue · comments
I think the actual implementation of CCT doesn't allow to pass non-square image.
If I pass img_size as a tuple I managed do it by changing these two lines
vit-pytorch/vit_pytorch/cct.py
Lines 327 to 328 in 2aae406
height=img_size[0],
width=img_size[1]),
Maybe you could modify the code in this repo in order that it works when img_size is an int or a tuple.
thx a lot !