youweiliang / evit

Python code for ICLR 2022 spotlight paper EViT: Expediting Vision Transformers via Token Reorganizations
Apache License 2.0
162 stars 19 forks source link

ImageNet version #20

Open victoriapf opened 1 year ago

victoriapf commented 1 year ago

Hello team, out of curiosity, from the text I'm imagining that you used the imagenet-21k to train the models, is that right? Or was it the imagenet-1k version?

Thanks!

youweiliang commented 1 year ago

Hi, Thanks for your interest in our paper! We used imagenet-1k for training and the imagenet validation set for reporting the accuracy.