Home

Awesome

EQ-Net: Elastic Quantization Neural Networks

Code for EQ-Net: Elastic Quantization Neural Networks accepted by ICCV2023

EQ-Net

Requirements

Run example

Config Preparation

We give a config sample under the configs folder.

Training

Train Supernet(Multi-GPU only)

python train_supernet.py --cfg cfg_path

Evaluation

Download the Trained Supermodel Google-Drive / BaiduDisk

Please fill in the trained Supernet model path in the configuration file, then run

python eval_supernet.py --cfg cfg_path

Thanks

Part of the code references pytorch-BigNAS and LSQplus. Please ref to their reposity for more details.