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

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

We give a config sample under the configs folder.

Train Supernet(Multi-GPU only)

python train_supernet.py --cfg cfg_path

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

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