/pytorch-mobilenet

Contains from-scratch implementation of the MobileNetV1, V2 and V3 paper with PyTorch. Each model architecture is contained in a single file for better portability & sharing.

Primary LanguagePythonMIT LicenseMIT

Stargazers