search

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

MobileViT use a larger GPU memory than AlexNet? #7

Closed anthonyweidai closed 11 months ago

anthonyweidai commented 2 years ago

I don't think mobile just have less parameters.