RuntimeError: shape '[0, 16, 3, 196, 196]' is invalid for input of size 460992 for swintiny

[Dongshengjiang]

Hello, I meet a error when I use local crop with size 96 for swin_tiny. is the released code not work for this size?

[ChunyuanLI]

Could you please provide more information? For example, are you using multi-crop? If yes, what are the number of crops and their crop size for global and local crops, respectively? and what is the input image size?

[Dongshengjiang]

Yes，when I use only global crops（224), it works. But when I set local crops >=1 with size 96, it shows the error.

[Dongshengjiang]

When I move the code to gpu, it seems there is no problem. Thanks!