├── .editorconfig ├── .gitignore ├── ACKNOWLEDGEMENTS ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── DATASETS.txt ├── LICENSE ├── README.md ├── coverage.ini ├── doc ├── Makefile ├── binary.rst ├── common.rst ├── conf.py ├── data.rst ├── index.rst ├── models.rst ├── quant.rst ├── release_notes.rst └── utils.rst ├── examples ├── cifar100 │ ├── cifar100.py │ ├── cifar100_fp.yaml │ ├── cifar100_ls1.yaml │ ├── cifar100_ls1_kd.yaml │ ├── cifar100_ls1_weight_fp_activation_kd.yaml │ ├── cifar100_ls1_weight_gf2_activation_kd.yaml │ ├── cifar100_ls1_weight_ls2_activation_kd.yaml │ └── cifar100_ls1_weight_lsT_activation_kd.yaml ├── imagenet │ ├── imagenet.py │ ├── imagenet_fp.yaml │ ├── imagenet_ls1_kd.yaml │ ├── imagenet_ls1_weight_fp_activation_kd.yaml │ ├── imagenet_ls1_weight_gf2_activation_kd.yaml │ ├── imagenet_ls1_weight_ls2_activation_kd.yaml │ └── imagenet_ls1_weight_lsT_activation_kd.yaml └── mnist │ ├── mnist.py │ ├── mnist_fp.yaml │ ├── mnist_ls1.yaml │ ├── mnist_ls1_weight_fp_activation.yaml │ ├── mnist_ls1_weight_gf2_activation.yaml │ ├── mnist_ls1_weight_ls2_activation.yaml │ └── mnist_ls1_weight_lsT_activation.yaml ├── mypy.ini ├── pyproject.toml ├── pytest.ini ├── quant ├── __init__.py ├── binary │ ├── __init__.py │ ├── activation_quantization.py │ ├── binary_conv.py │ ├── optimal.py │ ├── quantization.py │ ├── ste.py │ └── weight_quantization.py ├── common │ ├── __init__.py │ ├── compute_platform.py │ ├── experiment.py │ ├── initialization.py │ ├── metrics.py │ ├── parser.py │ ├── tasks.py │ └── training.py ├── data │ ├── __init__.py │ └── data_loaders.py ├── models │ ├── __init__.py │ ├── lenet.py │ └── resnet.py └── utils │ ├── __init__.py │ ├── checkpoints.py │ ├── kd_criterion.py │ ├── linear_lr_scheduler.py │ ├── moving_average.py │ ├── utils.py │ └── visualization.py ├── quant_logo.png ├── requirements.txt └── tests ├── __init__.py ├── binary ├── __init__.py ├── test_activation_quantization.py ├── test_binary_conv.py ├── test_quantization.py ├── test_ste.py └── test_weight_quantization.py ├── common ├── __init__.py ├── test_experiment.py ├── test_initialization.py ├── test_metrics.py ├── test_parser.py ├── test_tasks.py └── test_training.py ├── conftest.py ├── data ├── __init__.py ├── helpers.py └── test_data_loaders.py ├── models ├── __init__.py └── test_resnet.py └── utils ├── __init__.py ├── test_linear_lr_scheduler.py └── test_moving_average.py /.editorconfig: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/.editorconfig -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/.gitignore -------------------------------------------------------------------------------- /ACKNOWLEDGEMENTS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/ACKNOWLEDGEMENTS -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /DATASETS.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/DATASETS.txt -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/README.md -------------------------------------------------------------------------------- /coverage.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/coverage.ini -------------------------------------------------------------------------------- /doc/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/doc/Makefile -------------------------------------------------------------------------------- /doc/binary.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/doc/binary.rst -------------------------------------------------------------------------------- /doc/common.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/doc/common.rst -------------------------------------------------------------------------------- /doc/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/doc/conf.py -------------------------------------------------------------------------------- /doc/data.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/doc/data.rst -------------------------------------------------------------------------------- /doc/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/doc/index.rst -------------------------------------------------------------------------------- /doc/models.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/doc/models.rst -------------------------------------------------------------------------------- /doc/quant.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/doc/quant.rst -------------------------------------------------------------------------------- /doc/release_notes.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/doc/release_notes.rst -------------------------------------------------------------------------------- /doc/utils.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/doc/utils.rst -------------------------------------------------------------------------------- /examples/cifar100/cifar100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/examples/cifar100/cifar100.py -------------------------------------------------------------------------------- /examples/cifar100/cifar100_fp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/examples/cifar100/cifar100_fp.yaml -------------------------------------------------------------------------------- /examples/cifar100/cifar100_ls1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/examples/cifar100/cifar100_ls1.yaml -------------------------------------------------------------------------------- /examples/cifar100/cifar100_ls1_kd.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/examples/cifar100/cifar100_ls1_kd.yaml -------------------------------------------------------------------------------- /examples/cifar100/cifar100_ls1_weight_fp_activation_kd.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/examples/cifar100/cifar100_ls1_weight_fp_activation_kd.yaml -------------------------------------------------------------------------------- /examples/cifar100/cifar100_ls1_weight_gf2_activation_kd.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/examples/cifar100/cifar100_ls1_weight_gf2_activation_kd.yaml -------------------------------------------------------------------------------- /examples/cifar100/cifar100_ls1_weight_ls2_activation_kd.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/examples/cifar100/cifar100_ls1_weight_ls2_activation_kd.yaml -------------------------------------------------------------------------------- /examples/cifar100/cifar100_ls1_weight_lsT_activation_kd.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/examples/cifar100/cifar100_ls1_weight_lsT_activation_kd.yaml -------------------------------------------------------------------------------- /examples/imagenet/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/examples/imagenet/imagenet.py -------------------------------------------------------------------------------- /examples/imagenet/imagenet_fp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/examples/imagenet/imagenet_fp.yaml -------------------------------------------------------------------------------- /examples/imagenet/imagenet_ls1_kd.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/examples/imagenet/imagenet_ls1_kd.yaml -------------------------------------------------------------------------------- /examples/imagenet/imagenet_ls1_weight_fp_activation_kd.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/examples/imagenet/imagenet_ls1_weight_fp_activation_kd.yaml -------------------------------------------------------------------------------- /examples/imagenet/imagenet_ls1_weight_gf2_activation_kd.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/examples/imagenet/imagenet_ls1_weight_gf2_activation_kd.yaml -------------------------------------------------------------------------------- /examples/imagenet/imagenet_ls1_weight_ls2_activation_kd.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/examples/imagenet/imagenet_ls1_weight_ls2_activation_kd.yaml -------------------------------------------------------------------------------- /examples/imagenet/imagenet_ls1_weight_lsT_activation_kd.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/examples/imagenet/imagenet_ls1_weight_lsT_activation_kd.yaml -------------------------------------------------------------------------------- /examples/mnist/mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/examples/mnist/mnist.py -------------------------------------------------------------------------------- /examples/mnist/mnist_fp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/examples/mnist/mnist_fp.yaml -------------------------------------------------------------------------------- /examples/mnist/mnist_ls1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/examples/mnist/mnist_ls1.yaml -------------------------------------------------------------------------------- /examples/mnist/mnist_ls1_weight_fp_activation.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/examples/mnist/mnist_ls1_weight_fp_activation.yaml -------------------------------------------------------------------------------- /examples/mnist/mnist_ls1_weight_gf2_activation.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/examples/mnist/mnist_ls1_weight_gf2_activation.yaml -------------------------------------------------------------------------------- /examples/mnist/mnist_ls1_weight_ls2_activation.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/examples/mnist/mnist_ls1_weight_ls2_activation.yaml -------------------------------------------------------------------------------- /examples/mnist/mnist_ls1_weight_lsT_activation.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/examples/mnist/mnist_ls1_weight_lsT_activation.yaml -------------------------------------------------------------------------------- /mypy.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/mypy.ini -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/pyproject.toml -------------------------------------------------------------------------------- /pytest.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/pytest.ini -------------------------------------------------------------------------------- /quant/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/quant/__init__.py -------------------------------------------------------------------------------- /quant/binary/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/quant/binary/__init__.py -------------------------------------------------------------------------------- /quant/binary/activation_quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/quant/binary/activation_quantization.py -------------------------------------------------------------------------------- /quant/binary/binary_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/quant/binary/binary_conv.py -------------------------------------------------------------------------------- /quant/binary/optimal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/quant/binary/optimal.py -------------------------------------------------------------------------------- /quant/binary/quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/quant/binary/quantization.py -------------------------------------------------------------------------------- /quant/binary/ste.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/quant/binary/ste.py -------------------------------------------------------------------------------- /quant/binary/weight_quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/quant/binary/weight_quantization.py -------------------------------------------------------------------------------- /quant/common/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/quant/common/__init__.py -------------------------------------------------------------------------------- /quant/common/compute_platform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/quant/common/compute_platform.py -------------------------------------------------------------------------------- /quant/common/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/quant/common/experiment.py -------------------------------------------------------------------------------- /quant/common/initialization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/quant/common/initialization.py -------------------------------------------------------------------------------- /quant/common/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/quant/common/metrics.py -------------------------------------------------------------------------------- /quant/common/parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/quant/common/parser.py -------------------------------------------------------------------------------- /quant/common/tasks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/quant/common/tasks.py -------------------------------------------------------------------------------- /quant/common/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/quant/common/training.py -------------------------------------------------------------------------------- /quant/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/quant/data/__init__.py -------------------------------------------------------------------------------- /quant/data/data_loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/quant/data/data_loaders.py -------------------------------------------------------------------------------- /quant/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/quant/models/__init__.py -------------------------------------------------------------------------------- /quant/models/lenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/quant/models/lenet.py -------------------------------------------------------------------------------- /quant/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/quant/models/resnet.py -------------------------------------------------------------------------------- /quant/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/quant/utils/__init__.py -------------------------------------------------------------------------------- /quant/utils/checkpoints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/quant/utils/checkpoints.py -------------------------------------------------------------------------------- /quant/utils/kd_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/quant/utils/kd_criterion.py -------------------------------------------------------------------------------- /quant/utils/linear_lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/quant/utils/linear_lr_scheduler.py -------------------------------------------------------------------------------- /quant/utils/moving_average.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/quant/utils/moving_average.py -------------------------------------------------------------------------------- /quant/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/quant/utils/utils.py -------------------------------------------------------------------------------- /quant/utils/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/quant/utils/visualization.py -------------------------------------------------------------------------------- /quant_logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/quant_logo.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/requirements.txt -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/tests/__init__.py -------------------------------------------------------------------------------- /tests/binary/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/tests/binary/__init__.py -------------------------------------------------------------------------------- /tests/binary/test_activation_quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/tests/binary/test_activation_quantization.py -------------------------------------------------------------------------------- /tests/binary/test_binary_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/tests/binary/test_binary_conv.py -------------------------------------------------------------------------------- /tests/binary/test_quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/tests/binary/test_quantization.py -------------------------------------------------------------------------------- /tests/binary/test_ste.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/tests/binary/test_ste.py -------------------------------------------------------------------------------- /tests/binary/test_weight_quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/tests/binary/test_weight_quantization.py -------------------------------------------------------------------------------- /tests/common/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/tests/common/__init__.py -------------------------------------------------------------------------------- /tests/common/test_experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/tests/common/test_experiment.py -------------------------------------------------------------------------------- /tests/common/test_initialization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/tests/common/test_initialization.py -------------------------------------------------------------------------------- /tests/common/test_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/tests/common/test_metrics.py -------------------------------------------------------------------------------- /tests/common/test_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/tests/common/test_parser.py -------------------------------------------------------------------------------- /tests/common/test_tasks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/tests/common/test_tasks.py -------------------------------------------------------------------------------- /tests/common/test_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/tests/common/test_training.py -------------------------------------------------------------------------------- /tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/tests/conftest.py -------------------------------------------------------------------------------- /tests/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/tests/data/__init__.py -------------------------------------------------------------------------------- /tests/data/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/tests/data/helpers.py -------------------------------------------------------------------------------- /tests/data/test_data_loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/tests/data/test_data_loaders.py -------------------------------------------------------------------------------- /tests/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/tests/models/__init__.py -------------------------------------------------------------------------------- /tests/models/test_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/tests/models/test_resnet.py -------------------------------------------------------------------------------- /tests/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/tests/utils/__init__.py -------------------------------------------------------------------------------- /tests/utils/test_linear_lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/tests/utils/test_linear_lr_scheduler.py -------------------------------------------------------------------------------- /tests/utils/test_moving_average.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-quant/HEAD/tests/utils/test_moving_average.py --------------------------------------------------------------------------------