Skip to content

Latest commit

 

History

History
39 lines (24 loc) · 999 Bytes

README.md

File metadata and controls

39 lines (24 loc) · 999 Bytes
Raw

EQ-Net: Elastic Quantization Neural Networks

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

EQ-Net

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.