/mobilevit-pytorch

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

Primary LanguagePythonMIT LicenseMIT

Watchers