The inference speed is slow

chinhsuanwu / mobilevit-pytorch

A PyTorch implementation of "MobileViT: Light-weight, General-purpose, and Mobile-friendly Vision Transformer"

https://arxiv.org/abs/2110.02178

MIT License

501 stars 70 forks source link

Open ZhangLei999 opened 2 years ago

ZhangLei999 commented 2 years ago

Thanks for sharing your code! The inference speed of mobilevit is 40 ms/img. How to reproduce the speed in Table3?