microsoft / Cream

This is a collection of our NAS and Vision Transformer work.
MIT License
1.69k stars 230 forks source link

EfficientViT: code for downstream image classification #182

Closed jameslahm closed 1 year ago

jameslahm commented 1 year ago

Thank you for your great work! Would you mind sharing the code for downstream image classification in Table 4 in the paper? Thanks a lot!

xinyuliu-jeffrey commented 1 year ago

Hi @jameslahm ,

Sure, the training on downstream please follow the instructions for Imagenet classification. Please kindly use this file to build the downstream datasets ;)

Best, Xinyu

jameslahm commented 1 year ago

Thanks!