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

Error in sequence of steps in the model #12

Open hungryGeek16 opened 2 years ago

hungryGeek16 commented 2 years ago

@chinhsuanwu , thank you so much for uploading the code.