AssertionError: Input image size (32*32) doesn't match model (224*224).

[DDxk369]

Dear Author：
This error occurred when I tried to run resnet_vit_compare.py：
AssertionError: Input image size (3232) doesn't match model (224224).
Where do I need to modify it?

[AntixK]

Hi, resnet_vit_compare.py was an incomplete test code I used for some preliminary testing. I removed it now from the main repo.

To answer your question - the error is because ViT requires input image to be of size 224x224. So you can use the Imagenet dataset and resize it to 224x224 and compare the two models.