search

wilile26811249 / MobileViT

Unofficial PyTorch implementation of MobileViT based on paper "MobileViT: Light-weight, General-purpose, and Mobile-friendly Vision Transformer".
MIT License
112 stars 20 forks source link

Have you test MobileVit on cifar-10? #8

Open Jerryme-xxm opened 2 years ago

Jerryme-xxm commented 2 years ago

Thanks for your wonderful work！

I prepare to try MobileVit on small dataset, such as MNIST, and I need adjust the network structure. Before this work, I want to know if MobileVit has a better performance than other networks on small dataset.

I notice "get_cifar10_dataset" in utils.py. Have you tested MobileVit on cifar-10? If you have, could you please show me the accuracy and inference time result?

wilile26811249 commented 2 years ago

Ah... Sorry,I didn't test the performance of MobileVit on cifar-10.