amc
This repository offers a PyTorch implementation of techniques detailed in the paper 'AMC: AutoML for Model Compression and Acceleration on Mobile Devices'. It includes a methodological workflow for compressing MobileNet models on ImageNet, covering strategy search, weight export, and fine-tuning. The code enables replication of the compression process, facilitating significant FLOPs reduction without compromising accuracy. Pre-compressed MobileNet models are accessible in PyTorch and TensorFlow formats, alongside detailed performance statistics.
