EQ-Net: Elastic Quantization Neural Networks

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

Requirements

• Python = 3.8
• PyTorch = 1.13.0
• yacs = 0.1.8
• simplejson = 3.17.6

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.