Skip to content

xuke225/EQ-Net

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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.

About

EQ-Net [ICCV 2023]

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages