├── .gitignore ├── LICENSE ├── README.md ├── elastic_nn ├── modules │ ├── dynamic_layers.py │ └── dynamic_op.py ├── networks │ ├── dynamic_proxyless.py │ └── dynamic_quantized_proxyless.py └── utils.py ├── figures ├── cost.png ├── joint.png ├── overview.png └── seq.png ├── imagenet_codebase ├── data_providers │ ├── base_provider.py │ ├── coil.py │ ├── hand.py │ ├── imagenet.py │ └── svhn.py ├── modules │ └── layers.py ├── networks │ ├── __init__.py │ └── proxyless_nets.py ├── run_manager │ ├── __init__.py │ └── run_manager.py └── utils │ ├── __init__.py │ ├── flops_counter.py │ ├── latency_estimator.py │ ├── my_modules.py │ ├── pytorch_modules.py │ ├── pytorch_utils.py │ └── quantize_utils.py ├── lut └── BitFusion_new.b1.dict ├── methods └── evolution │ ├── evo_main_gather.py │ └── evolution_finder.py ├── quant_aware.py ├── search.py ├── test.py └── utils ├── accuracy_predictor.py ├── converter.py └── latency_predictor.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/README.md -------------------------------------------------------------------------------- /elastic_nn/modules/dynamic_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/elastic_nn/modules/dynamic_layers.py -------------------------------------------------------------------------------- /elastic_nn/modules/dynamic_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/elastic_nn/modules/dynamic_op.py -------------------------------------------------------------------------------- /elastic_nn/networks/dynamic_proxyless.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/elastic_nn/networks/dynamic_proxyless.py -------------------------------------------------------------------------------- /elastic_nn/networks/dynamic_quantized_proxyless.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/elastic_nn/networks/dynamic_quantized_proxyless.py -------------------------------------------------------------------------------- /elastic_nn/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/elastic_nn/utils.py -------------------------------------------------------------------------------- /figures/cost.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/figures/cost.png -------------------------------------------------------------------------------- /figures/joint.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/figures/joint.png -------------------------------------------------------------------------------- /figures/overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/figures/overview.png -------------------------------------------------------------------------------- /figures/seq.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/figures/seq.png -------------------------------------------------------------------------------- /imagenet_codebase/data_providers/base_provider.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/imagenet_codebase/data_providers/base_provider.py -------------------------------------------------------------------------------- /imagenet_codebase/data_providers/coil.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/imagenet_codebase/data_providers/coil.py -------------------------------------------------------------------------------- /imagenet_codebase/data_providers/hand.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/imagenet_codebase/data_providers/hand.py -------------------------------------------------------------------------------- /imagenet_codebase/data_providers/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/imagenet_codebase/data_providers/imagenet.py -------------------------------------------------------------------------------- /imagenet_codebase/data_providers/svhn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/imagenet_codebase/data_providers/svhn.py -------------------------------------------------------------------------------- /imagenet_codebase/modules/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/imagenet_codebase/modules/layers.py -------------------------------------------------------------------------------- /imagenet_codebase/networks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/imagenet_codebase/networks/__init__.py -------------------------------------------------------------------------------- /imagenet_codebase/networks/proxyless_nets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/imagenet_codebase/networks/proxyless_nets.py -------------------------------------------------------------------------------- /imagenet_codebase/run_manager/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/imagenet_codebase/run_manager/__init__.py -------------------------------------------------------------------------------- /imagenet_codebase/run_manager/run_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/imagenet_codebase/run_manager/run_manager.py -------------------------------------------------------------------------------- /imagenet_codebase/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/imagenet_codebase/utils/__init__.py -------------------------------------------------------------------------------- /imagenet_codebase/utils/flops_counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/imagenet_codebase/utils/flops_counter.py -------------------------------------------------------------------------------- /imagenet_codebase/utils/latency_estimator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/imagenet_codebase/utils/latency_estimator.py -------------------------------------------------------------------------------- /imagenet_codebase/utils/my_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/imagenet_codebase/utils/my_modules.py -------------------------------------------------------------------------------- /imagenet_codebase/utils/pytorch_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/imagenet_codebase/utils/pytorch_modules.py -------------------------------------------------------------------------------- /imagenet_codebase/utils/pytorch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/imagenet_codebase/utils/pytorch_utils.py -------------------------------------------------------------------------------- /imagenet_codebase/utils/quantize_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/imagenet_codebase/utils/quantize_utils.py -------------------------------------------------------------------------------- /lut/BitFusion_new.b1.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/lut/BitFusion_new.b1.dict -------------------------------------------------------------------------------- /methods/evolution/evo_main_gather.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/methods/evolution/evo_main_gather.py -------------------------------------------------------------------------------- /methods/evolution/evolution_finder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/methods/evolution/evolution_finder.py -------------------------------------------------------------------------------- /quant_aware.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/quant_aware.py -------------------------------------------------------------------------------- /search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/search.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/test.py -------------------------------------------------------------------------------- /utils/accuracy_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/utils/accuracy_predictor.py -------------------------------------------------------------------------------- /utils/converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/utils/converter.py -------------------------------------------------------------------------------- /utils/latency_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/apq/HEAD/utils/latency_predictor.py --------------------------------------------------------------------------------