MnasNet-PyTorch icon indicating copy to clipboard operation
MnasNet-PyTorch copied to clipboard

A PyTorch implementation of Mnasnet: MnasNet: Platform-Aware Neural Architecture Search for Mobile.

Overview

A PyTorch implementation of Mnasnet searched architecture: MnasNet: Platform-Aware Neural Architecture Search for Mobile. On the ImageNet classification task, the model achieves 74.0% top-1 accuracy with 76ms latency on a Pixel phone, which is 1.5× faster than MobileNetV2

MnasNet Architecture

Alt text

Disclaimer

Codes modified from mobilenet-v2