├── .gitignore ├── .gitmodules ├── ILP.ipynb ├── LICENSE ├── README.md ├── bit_config.py ├── imgs └── resnet18_TC.png ├── model_zoo.md ├── quant_train.py ├── requirements.txt ├── tvm_benchmark ├── README.md ├── hawq_utils_resnet50.py ├── mixed_precision_models │ ├── __init__.py │ ├── init.py │ ├── layers.py │ ├── mobilenet_v2.py │ ├── quantized_inception.py │ ├── quantized_resnet_v1.py │ ├── quantized_resnet_v1_5.py │ ├── quantized_resnet_v2.py │ └── tuning_logs │ │ ├── resnet18_HWNC_mixed_batch_8.log │ │ └── resnet50_HWNC_mixed_batch_8.log ├── models │ └── input_image_batch_1.npy ├── run_resnet_inference_time.sh ├── test_resnet_accuracy_imagenet.py ├── test_resnet_inference.py └── test_resnet_inference_time.py └── utils ├── __init__.py ├── data_utils.py ├── export ├── README.md ├── __init__.py ├── export_modules.py ├── export_utils.py ├── function.py └── manager.py ├── models ├── q_inceptionv3.py ├── q_mobilenetv2.py └── q_resnet.py └── quantization_utils ├── quant_modules.py └── quant_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/.gitmodules -------------------------------------------------------------------------------- /ILP.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/ILP.ipynb -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/README.md -------------------------------------------------------------------------------- /bit_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/bit_config.py -------------------------------------------------------------------------------- /imgs/resnet18_TC.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/imgs/resnet18_TC.png -------------------------------------------------------------------------------- /model_zoo.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/model_zoo.md -------------------------------------------------------------------------------- /quant_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/quant_train.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/requirements.txt -------------------------------------------------------------------------------- /tvm_benchmark/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/tvm_benchmark/README.md -------------------------------------------------------------------------------- /tvm_benchmark/hawq_utils_resnet50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/tvm_benchmark/hawq_utils_resnet50.py -------------------------------------------------------------------------------- /tvm_benchmark/mixed_precision_models/__init__.py: -------------------------------------------------------------------------------- 1 | from . import layers -------------------------------------------------------------------------------- /tvm_benchmark/mixed_precision_models/init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/tvm_benchmark/mixed_precision_models/init.py -------------------------------------------------------------------------------- /tvm_benchmark/mixed_precision_models/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/tvm_benchmark/mixed_precision_models/layers.py -------------------------------------------------------------------------------- /tvm_benchmark/mixed_precision_models/mobilenet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/tvm_benchmark/mixed_precision_models/mobilenet_v2.py -------------------------------------------------------------------------------- /tvm_benchmark/mixed_precision_models/quantized_inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/tvm_benchmark/mixed_precision_models/quantized_inception.py -------------------------------------------------------------------------------- /tvm_benchmark/mixed_precision_models/quantized_resnet_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/tvm_benchmark/mixed_precision_models/quantized_resnet_v1.py -------------------------------------------------------------------------------- /tvm_benchmark/mixed_precision_models/quantized_resnet_v1_5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/tvm_benchmark/mixed_precision_models/quantized_resnet_v1_5.py -------------------------------------------------------------------------------- /tvm_benchmark/mixed_precision_models/quantized_resnet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/tvm_benchmark/mixed_precision_models/quantized_resnet_v2.py -------------------------------------------------------------------------------- /tvm_benchmark/mixed_precision_models/tuning_logs/resnet18_HWNC_mixed_batch_8.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/tvm_benchmark/mixed_precision_models/tuning_logs/resnet18_HWNC_mixed_batch_8.log -------------------------------------------------------------------------------- /tvm_benchmark/mixed_precision_models/tuning_logs/resnet50_HWNC_mixed_batch_8.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/tvm_benchmark/mixed_precision_models/tuning_logs/resnet50_HWNC_mixed_batch_8.log -------------------------------------------------------------------------------- /tvm_benchmark/models/input_image_batch_1.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/tvm_benchmark/models/input_image_batch_1.npy -------------------------------------------------------------------------------- /tvm_benchmark/run_resnet_inference_time.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/tvm_benchmark/run_resnet_inference_time.sh -------------------------------------------------------------------------------- /tvm_benchmark/test_resnet_accuracy_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/tvm_benchmark/test_resnet_accuracy_imagenet.py -------------------------------------------------------------------------------- /tvm_benchmark/test_resnet_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/tvm_benchmark/test_resnet_inference.py -------------------------------------------------------------------------------- /tvm_benchmark/test_resnet_inference_time.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/tvm_benchmark/test_resnet_inference_time.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/utils/data_utils.py -------------------------------------------------------------------------------- /utils/export/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/utils/export/README.md -------------------------------------------------------------------------------- /utils/export/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/utils/export/__init__.py -------------------------------------------------------------------------------- /utils/export/export_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/utils/export/export_modules.py -------------------------------------------------------------------------------- /utils/export/export_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/utils/export/export_utils.py -------------------------------------------------------------------------------- /utils/export/function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/utils/export/function.py -------------------------------------------------------------------------------- /utils/export/manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/utils/export/manager.py -------------------------------------------------------------------------------- /utils/models/q_inceptionv3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/utils/models/q_inceptionv3.py -------------------------------------------------------------------------------- /utils/models/q_mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/utils/models/q_mobilenetv2.py -------------------------------------------------------------------------------- /utils/models/q_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/utils/models/q_resnet.py -------------------------------------------------------------------------------- /utils/quantization_utils/quant_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/utils/quantization_utils/quant_modules.py -------------------------------------------------------------------------------- /utils/quantization_utils/quant_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zhen-Dong/HAWQ/HEAD/utils/quantization_utils/quant_utils.py --------------------------------------------------------------------------------