PyPI - mct-nightly - Versions diffs - 1.7.1.31122022.post351__tar.gz → 1.8.0.1042023.post423__tar.gz - Mend

mct-nightly 1.7.1.31122022.post351tar.gz → 1.8.0.1042023.post423tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (540) hide show

{mct-nightly-1.7.1.31122022.post351 → mct-nightly-1.8.0.1042023.post423}/PKG-INFO RENAMED Viewed

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mct-nightly
-Version: 1.7.1.31122022.post351
+Version: 1.8.0.1042023.post423
 Summary: A Model Compression Toolkit for neural networks
 Home-page: UNKNOWN
 License: UNKNOWN
@@ -29,8 +29,8 @@ Description: # Model Compression Toolkit (MCT)
         ## Supported Features
         MCT supports different quantization methods:
-        * Post training quantization (PTQ): [Keras API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/keras_post_training_quantization_experimental.html#ug-keras-post-training-quantization-experimental), [PyTorch API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/pytorch_post_training_quantization_experimental.html#ug-pytorch-post-training-quantization-experimental)
-        * Gradient-based post training quantization (GPTQ): [Keras API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/keras_gradient_post_training_quantization_experimental.html#ug-keras-gradient-post-training-quantization-experimental), [PyTorch API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/pytorch_gradient_post_training_quantization_experimental.html#ug-pytorch-gradient-post-training-quantization-experimental)
+        * Post-training quantization (PTQ): [Keras API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/keras_post_training_quantization_experimental.html#ug-keras-post-training-quantization-experimental), [PyTorch API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/pytorch_post_training_quantization_experimental.html#ug-pytorch-post-training-quantization-experimental)
+        * Gradient-based post-training quantization (GPTQ): [Keras API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/keras_gradient_post_training_quantization_experimental.html#ug-keras-gradient-post-training-quantization-experimental), [PyTorch API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/pytorch_gradient_post_training_quantization_experimental.html#ug-pytorch-gradient-post-training-quantization-experimental)
         * Quantization aware training (QAT)[*](#experimental-features)
@@ -87,15 +87,15 @@ Description: # Model Compression Toolkit (MCT)
         pip install mct-nightly
         ```
-        ### Requierments
+        ### Requirements
-        To run MCT, one of the supported frameworks, Tenosflow/Pytorch, needs to be installed.
+        To run MCT, one of the supported frameworks, Tensorflow/Pytorch, needs to be installed.
-        For using with Tensorflow please install the packages:
+        For use with Tensorflow please install the packages:
         [tensorflow](https://www.tensorflow.org/install),
         [tensorflow-model-optimization](https://www.tensorflow.org/model_optimization/guide/install)
-        For using with PyTorch please install the packages:
+        For use with PyTorch please install the packages:
         [torch](https://pytorch.org/)
         Also, a [requirements](requirements.txt) file can be used to set up your environment.
@@ -107,12 +107,12 @@ Description: # Model Compression Toolkit (MCT)
-        | Python Version                                                                                                                                                                                                                                                             |
-        |----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
-        | [![Run Tests - Python 3.10](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_all_latest_frameworks_python310.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_all_latest_frameworks_python310.yml) |
-        | [![Run Tests - Python 3.9](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_all_latest_frameworks_python39.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_all_latest_frameworks_python39.yml)    |
-        | [![Run Tests - Python 3.8](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_all_latest_frameworks_python38.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_all_latest_frameworks_python38.yml)    |
-        | [![Run Tests - Python 3.7](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_all_latest_frameworks_python37.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_all_latest_frameworks_python37.yml)    |
+        | Python Version                                                                                                                                                                                                                                                            |
+        |---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
+        | [![Run Tests - Python 3.10](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_python310.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_python310.yml) |
+        | [![Run Tests - Python 3.9](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_python39.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_python39.yml)   |
+        | [![Run Tests - Python 3.8](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_python38.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_python38.yml)   |
+        | [![Run Tests - Python 3.7](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_python37.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_python37.yml)   |
         ### Supported NN-Frameworks Versions
@@ -122,9 +122,9 @@ Description: # Model Compression Toolkit (MCT)
         | TensorFlow Version                                                                                   | PyTorch Version                                                                                          |
         |------------------------------------------------------------------------------------------------------|----------------------------------------------------------------------------------------------------------|
-        | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_tf210.yml/badge.svg) | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_torch1_13.yml/badge.svg) |
-        | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_tf29.yml/badge.svg)  | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_torch1_12.yml/badge.svg) |
-        | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_tf28.yml/badge.svg)  | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_torch1_11.yml/badge.svg) |
+        | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_tf211.yml/badge.svg) | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_torch1_13.yml/badge.svg) |
+        | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_tf210.yml/badge.svg) | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_torch1_12.yml/badge.svg) |
+        | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_tf29.yml/badge.svg)  | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_torch1_11.yml/badge.svg) |
         ### Usage Example

{mct-nightly-1.7.1.31122022.post351 → mct-nightly-1.8.0.1042023.post423}/README.md RENAMED Viewed

@@ -23,8 +23,8 @@ MCT is developed by researchers and engineers working at Sony Semiconductor Isra
 ## Supported Features
 MCT supports different quantization methods:
-* Post training quantization (PTQ): [Keras API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/keras_post_training_quantization_experimental.html#ug-keras-post-training-quantization-experimental), [PyTorch API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/pytorch_post_training_quantization_experimental.html#ug-pytorch-post-training-quantization-experimental)
-* Gradient-based post training quantization (GPTQ): [Keras API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/keras_gradient_post_training_quantization_experimental.html#ug-keras-gradient-post-training-quantization-experimental), [PyTorch API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/pytorch_gradient_post_training_quantization_experimental.html#ug-pytorch-gradient-post-training-quantization-experimental)
+* Post-training quantization (PTQ): [Keras API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/keras_post_training_quantization_experimental.html#ug-keras-post-training-quantization-experimental), [PyTorch API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/pytorch_post_training_quantization_experimental.html#ug-pytorch-post-training-quantization-experimental)
+* Gradient-based post-training quantization (GPTQ): [Keras API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/keras_gradient_post_training_quantization_experimental.html#ug-keras-gradient-post-training-quantization-experimental), [PyTorch API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/pytorch_gradient_post_training_quantization_experimental.html#ug-pytorch-gradient-post-training-quantization-experimental)
 * Quantization aware training (QAT)[*](#experimental-features)
@@ -81,15 +81,15 @@ A nightly package is also available (unstable):
 pip install mct-nightly
 ```
-### Requierments
+### Requirements
-To run MCT, one of the supported frameworks, Tenosflow/Pytorch, needs to be installed.
+To run MCT, one of the supported frameworks, Tensorflow/Pytorch, needs to be installed.
-For using with Tensorflow please install the packages:
+For use with Tensorflow please install the packages:
 [tensorflow](https://www.tensorflow.org/install),
 [tensorflow-model-optimization](https://www.tensorflow.org/model_optimization/guide/install)
-For using with PyTorch please install the packages:
+For use with PyTorch please install the packages:
 [torch](https://pytorch.org/)
 Also, a [requirements](requirements.txt) file can be used to set up your environment.
@@ -101,12 +101,12 @@ Currently, MCT is being tested on various Python versions:
-| Python Version                                                                                                                                                                                                                                                             |
-|----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
-| [![Run Tests - Python 3.10](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_all_latest_frameworks_python310.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_all_latest_frameworks_python310.yml) |
-| [![Run Tests - Python 3.9](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_all_latest_frameworks_python39.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_all_latest_frameworks_python39.yml)    |
-| [![Run Tests - Python 3.8](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_all_latest_frameworks_python38.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_all_latest_frameworks_python38.yml)    |
-| [![Run Tests - Python 3.7](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_all_latest_frameworks_python37.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_all_latest_frameworks_python37.yml)    |
+| Python Version                                                                                                                                                                                                                                                            |
+|---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
+| [![Run Tests - Python 3.10](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_python310.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_python310.yml) |
+| [![Run Tests - Python 3.9](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_python39.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_python39.yml)   |
+| [![Run Tests - Python 3.8](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_python38.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_python38.yml)   |
+| [![Run Tests - Python 3.7](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_python37.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_python37.yml)   |
 ### Supported NN-Frameworks Versions
@@ -116,9 +116,9 @@ is tested on various versions:
 | TensorFlow Version                                                                                   | PyTorch Version                                                                                          |
 |------------------------------------------------------------------------------------------------------|----------------------------------------------------------------------------------------------------------|
-| ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_tf210.yml/badge.svg) | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_torch1_13.yml/badge.svg) |
-| ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_tf29.yml/badge.svg)  | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_torch1_12.yml/badge.svg) |
-| ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_tf28.yml/badge.svg)  | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_torch1_11.yml/badge.svg) |
+| ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_tf211.yml/badge.svg) | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_torch1_13.yml/badge.svg) |
+| ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_tf210.yml/badge.svg) | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_torch1_12.yml/badge.svg) |
+| ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_tf29.yml/badge.svg)  | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_torch1_11.yml/badge.svg) |
 ### Usage Example

{mct-nightly-1.7.1.31122022.post351 → mct-nightly-1.8.0.1042023.post423}/mct_nightly.egg-info/PKG-INFO RENAMED Viewed

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mct-nightly
-Version: 1.7.1.31122022.post351
+Version: 1.8.0.1042023.post423
 Summary: A Model Compression Toolkit for neural networks
 Home-page: UNKNOWN
 License: UNKNOWN
@@ -29,8 +29,8 @@ Description: # Model Compression Toolkit (MCT)
         ## Supported Features
         MCT supports different quantization methods:
-        * Post training quantization (PTQ): [Keras API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/keras_post_training_quantization_experimental.html#ug-keras-post-training-quantization-experimental), [PyTorch API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/pytorch_post_training_quantization_experimental.html#ug-pytorch-post-training-quantization-experimental)
-        * Gradient-based post training quantization (GPTQ): [Keras API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/keras_gradient_post_training_quantization_experimental.html#ug-keras-gradient-post-training-quantization-experimental), [PyTorch API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/pytorch_gradient_post_training_quantization_experimental.html#ug-pytorch-gradient-post-training-quantization-experimental)
+        * Post-training quantization (PTQ): [Keras API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/keras_post_training_quantization_experimental.html#ug-keras-post-training-quantization-experimental), [PyTorch API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/pytorch_post_training_quantization_experimental.html#ug-pytorch-post-training-quantization-experimental)
+        * Gradient-based post-training quantization (GPTQ): [Keras API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/keras_gradient_post_training_quantization_experimental.html#ug-keras-gradient-post-training-quantization-experimental), [PyTorch API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/pytorch_gradient_post_training_quantization_experimental.html#ug-pytorch-gradient-post-training-quantization-experimental)
         * Quantization aware training (QAT)[*](#experimental-features)
@@ -87,15 +87,15 @@ Description: # Model Compression Toolkit (MCT)
         pip install mct-nightly
         ```
-        ### Requierments
+        ### Requirements
-        To run MCT, one of the supported frameworks, Tenosflow/Pytorch, needs to be installed.
+        To run MCT, one of the supported frameworks, Tensorflow/Pytorch, needs to be installed.
-        For using with Tensorflow please install the packages:
+        For use with Tensorflow please install the packages:
         [tensorflow](https://www.tensorflow.org/install),
         [tensorflow-model-optimization](https://www.tensorflow.org/model_optimization/guide/install)
-        For using with PyTorch please install the packages:
+        For use with PyTorch please install the packages:
         [torch](https://pytorch.org/)
         Also, a [requirements](requirements.txt) file can be used to set up your environment.
@@ -107,12 +107,12 @@ Description: # Model Compression Toolkit (MCT)
-        | Python Version                                                                                                                                                                                                                                                             |
-        |----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
-        | [![Run Tests - Python 3.10](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_all_latest_frameworks_python310.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_all_latest_frameworks_python310.yml) |
-        | [![Run Tests - Python 3.9](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_all_latest_frameworks_python39.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_all_latest_frameworks_python39.yml)    |
-        | [![Run Tests - Python 3.8](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_all_latest_frameworks_python38.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_all_latest_frameworks_python38.yml)    |
-        | [![Run Tests - Python 3.7](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_all_latest_frameworks_python37.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_all_latest_frameworks_python37.yml)    |
+        | Python Version                                                                                                                                                                                                                                                            |
+        |---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
+        | [![Run Tests - Python 3.10](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_python310.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_python310.yml) |
+        | [![Run Tests - Python 3.9](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_python39.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_python39.yml)   |
+        | [![Run Tests - Python 3.8](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_python38.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_python38.yml)   |
+        | [![Run Tests - Python 3.7](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_python37.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite_python37.yml)   |
         ### Supported NN-Frameworks Versions
@@ -122,9 +122,9 @@ Description: # Model Compression Toolkit (MCT)
         | TensorFlow Version                                                                                   | PyTorch Version                                                                                          |
         |------------------------------------------------------------------------------------------------------|----------------------------------------------------------------------------------------------------------|
-        | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_tf210.yml/badge.svg) | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_torch1_13.yml/badge.svg) |
-        | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_tf29.yml/badge.svg)  | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_torch1_12.yml/badge.svg) |
-        | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_tf28.yml/badge.svg)  | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_torch1_11.yml/badge.svg) |
+        | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_tf211.yml/badge.svg) | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_torch1_13.yml/badge.svg) |
+        | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_tf210.yml/badge.svg) | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_torch1_12.yml/badge.svg) |
+        | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_tf29.yml/badge.svg)  | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_torch1_11.yml/badge.svg) |
         ### Usage Example

{mct-nightly-1.7.1.31122022.post351 → mct-nightly-1.8.0.1042023.post423}/mct_nightly.egg-info/SOURCES.txt RENAMED Viewed

@@ -288,6 +288,13 @@ model_compression_toolkit/core/tpc_models/default_tpc/v5/__init__.py
 model_compression_toolkit/core/tpc_models/default_tpc/v5/tp_model.py
 model_compression_toolkit/core/tpc_models/default_tpc/v5/tpc_keras.py
 model_compression_toolkit/core/tpc_models/default_tpc/v5/tpc_pytorch.py
+model_compression_toolkit/core/tpc_models/imx500_tpc/__init__.py
+model_compression_toolkit/core/tpc_models/imx500_tpc/target_platform_capabilities.py
+model_compression_toolkit/core/tpc_models/imx500_tpc/latest/__init__.py
+model_compression_toolkit/core/tpc_models/imx500_tpc/v1/__init__.py
+model_compression_toolkit/core/tpc_models/imx500_tpc/v1/tp_model.py
+model_compression_toolkit/core/tpc_models/imx500_tpc/v1/tpc_keras.py
+model_compression_toolkit/core/tpc_models/imx500_tpc/v1/tpc_pytorch.py
 model_compression_toolkit/core/tpc_models/qnnpack_tpc/__init__.py
 model_compression_toolkit/core/tpc_models/qnnpack_tpc/target_platform_capabilities.py
 model_compression_toolkit/core/tpc_models/qnnpack_tpc/latest/__init__.py
@@ -317,35 +324,21 @@ model_compression_toolkit/exporter/model_exporter/pytorch/fakely_quant_torchscri
 model_compression_toolkit/exporter/model_exporter/pytorch/pytorch_export_facade.py
 model_compression_toolkit/exporter/model_exporter/tflite/__init__.py
 model_compression_toolkit/exporter/model_exporter/tflite/fakely_quant_tflite_exporter.py
+model_compression_toolkit/exporter/model_exporter/tflite/int8_tflite_exporter.py
 model_compression_toolkit/exporter/model_exporter/tflite/tflite_export_facade.py
 model_compression_toolkit/exporter/model_wrapper/__init__.py
 model_compression_toolkit/exporter/model_wrapper/keras/__init__.py
-model_compression_toolkit/exporter/model_wrapper/keras/extended_quantize_wrapper.py
 model_compression_toolkit/exporter/model_wrapper/keras/validate_layer.py
 model_compression_toolkit/exporter/model_wrapper/keras/builder/__init__.py
 model_compression_toolkit/exporter/model_wrapper/keras/builder/fully_quantized_model_builder.py
-model_compression_toolkit/exporter/model_wrapper/keras/builder/quantize_config_to_node.py
-model_compression_toolkit/exporter/model_wrapper/keras/builder/quantizer_to_node.py
-model_compression_toolkit/exporter/model_wrapper/keras/quantize_configs/__init__.py
-model_compression_toolkit/exporter/model_wrapper/keras/quantize_configs/activation_quantize_config.py
-model_compression_toolkit/exporter/model_wrapper/keras/quantize_configs/weights_activation_quantize_config.py
-model_compression_toolkit/exporter/model_wrapper/keras/quantize_configs/weights_quantize_config.py
-model_compression_toolkit/exporter/model_wrapper/keras/quantizers/__init__.py
-model_compression_toolkit/exporter/model_wrapper/keras/quantizers/fq_quantizer.py
-model_compression_toolkit/exporter/model_wrapper/keras/quantizers/weights_uniform_quantizer.py
+model_compression_toolkit/exporter/model_wrapper/keras/builder/node_to_quantizer.py
+model_compression_toolkit/exporter/model_wrapper/keras/builder/node_to_quantizers.py
 model_compression_toolkit/exporter/model_wrapper/pytorch/__init__.py
+model_compression_toolkit/exporter/model_wrapper/pytorch/validate_layer.py
 model_compression_toolkit/exporter/model_wrapper/pytorch/builder/__init__.py
 model_compression_toolkit/exporter/model_wrapper/pytorch/builder/fully_quantized_model_builder.py
-model_compression_toolkit/exporter/model_wrapper/pytorch/builder/node_to_quantize_config.py
 model_compression_toolkit/exporter/model_wrapper/pytorch/builder/node_to_quantizer.py
-model_compression_toolkit/exporter/model_wrapper/pytorch/quantizers/__init__.py
-model_compression_toolkit/exporter/model_wrapper/pytorch/quantizers/fq_quantizer.py
-model_compression_toolkit/exporter/model_wrapper/pytorch/quantizers/uniform_weights_quantizer.py
-model_compression_toolkit/exporter/model_wrapper/pytorch/wrappers_quantize_configs/__init__.py
-model_compression_toolkit/exporter/model_wrapper/pytorch/wrappers_quantize_configs/activation_quantize_config.py
-model_compression_toolkit/exporter/model_wrapper/pytorch/wrappers_quantize_configs/no_quantization_quantize_config.py
-model_compression_toolkit/exporter/model_wrapper/pytorch/wrappers_quantize_configs/weights_activation_quantize_config.py
-model_compression_toolkit/exporter/model_wrapper/pytorch/wrappers_quantize_configs/weights_quantize_config.py
+model_compression_toolkit/exporter/model_wrapper/pytorch/builder/node_to_quantizers.py
 model_compression_toolkit/gptq/__init__.py
 model_compression_toolkit/gptq/runner.py
 model_compression_toolkit/gptq/common/__init__.py
@@ -355,43 +348,35 @@ model_compression_toolkit/gptq/common/gptq_graph.py
 model_compression_toolkit/gptq/common/gptq_training.py
 model_compression_toolkit/gptq/keras/__init__.py
 model_compression_toolkit/gptq/keras/gptq_loss.py
-model_compression_toolkit/gptq/keras/gptq_model_builder.py
 model_compression_toolkit/gptq/keras/gptq_training.py
 model_compression_toolkit/gptq/keras/graph_info.py
 model_compression_toolkit/gptq/keras/quantization_facade.py
-model_compression_toolkit/gptq/keras/optimizers/__init__.py
-model_compression_toolkit/gptq/keras/optimizers/sam_optimizer.py
 model_compression_toolkit/gptq/keras/quantizer/__init__.py
-model_compression_toolkit/gptq/keras/quantizer/config_factory.py
-model_compression_toolkit/gptq/keras/quantizer/kernel_functions.py
+model_compression_toolkit/gptq/keras/quantizer/base_keras_gptq_quantizer.py
 model_compression_toolkit/gptq/keras/quantizer/quant_utils.py
-model_compression_toolkit/gptq/keras/quantizer/configs/__init__.py
-model_compression_toolkit/gptq/keras/quantizer/configs/base_quantizer_gptq_config.py
-model_compression_toolkit/gptq/keras/quantizer/configs/weight_quantizer_gptq_config.py
-model_compression_toolkit/gptq/keras/quantizer/gumbel_rounding/__init__.py
-model_compression_toolkit/gptq/keras/quantizer/gumbel_rounding/base_gumbel_rounding.py
-model_compression_toolkit/gptq/keras/quantizer/gumbel_rounding/gumbel_softmax.py
-model_compression_toolkit/gptq/keras/quantizer/gumbel_rounding/symmetric_gumbel.py
-model_compression_toolkit/gptq/keras/quantizer/gumbel_rounding/uniform_gumbel.py
+model_compression_toolkit/gptq/keras/quantizer/quantization_builder.py
+model_compression_toolkit/gptq/keras/quantizer/regularization_factory.py
+model_compression_toolkit/gptq/keras/quantizer/soft_rounding/__init__.py
+model_compression_toolkit/gptq/keras/quantizer/soft_rounding/soft_quantizer_reg.py
+model_compression_toolkit/gptq/keras/quantizer/soft_rounding/symmetric_soft_quantizer.py
 model_compression_toolkit/gptq/keras/quantizer/ste_rounding/__init__.py
 model_compression_toolkit/gptq/keras/quantizer/ste_rounding/symmetric_ste.py
-model_compression_toolkit/gptq/keras/quantizer/ste_rounding/uniform_ste.py
 model_compression_toolkit/gptq/pytorch/__init__.py
-model_compression_toolkit/gptq/pytorch/gptq_graph_info.py
 model_compression_toolkit/gptq/pytorch/gptq_loss.py
-model_compression_toolkit/gptq/pytorch/gptq_model_builder.py
 model_compression_toolkit/gptq/pytorch/gptq_training.py
+model_compression_toolkit/gptq/pytorch/graph_info.py
 model_compression_toolkit/gptq/pytorch/quantization_facade.py
 model_compression_toolkit/gptq/pytorch/quantizer/__init__.py
-model_compression_toolkit/gptq/pytorch/quantizer/gptq_quantizer.py
+model_compression_toolkit/gptq/pytorch/quantizer/base_pytorch_gptq_quantizer.py
 model_compression_toolkit/gptq/pytorch/quantizer/quant_utils.py
-model_compression_toolkit/gptq/pytorch/quantizer/quantizer_wrapper.py
-model_compression_toolkit/gptq/pytorch/quantizer/gumbel_rounding/__init__.py
-model_compression_toolkit/gptq/pytorch/quantizer/gumbel_rounding/base_gumbel_weights_quantizer.py
-model_compression_toolkit/gptq/pytorch/quantizer/gumbel_rounding/sym_gumbel_weights_quantizer.py
-model_compression_toolkit/gptq/pytorch/quantizer/gumbel_rounding/uniform_gumbel_weights_quantizer.py
+model_compression_toolkit/gptq/pytorch/quantizer/quantization_builder.py
+model_compression_toolkit/gptq/pytorch/quantizer/regularization_factory.py
+model_compression_toolkit/gptq/pytorch/quantizer/soft_rounding/__init__.py
+model_compression_toolkit/gptq/pytorch/quantizer/soft_rounding/soft_quantizer_reg.py
+model_compression_toolkit/gptq/pytorch/quantizer/soft_rounding/symmetric_soft_quantizer.py
+model_compression_toolkit/gptq/pytorch/quantizer/soft_rounding/uniform_soft_quantizer.py
 model_compression_toolkit/gptq/pytorch/quantizer/ste_rounding/__init__.py
-model_compression_toolkit/gptq/pytorch/quantizer/ste_rounding/ste_weights_quantizer.py
+model_compression_toolkit/gptq/pytorch/quantizer/ste_rounding/symmetric_ste.py
 model_compression_toolkit/ptq/__init__.py
 model_compression_toolkit/ptq/runner.py
 model_compression_toolkit/ptq/keras/__init__.py
@@ -401,24 +386,82 @@ model_compression_toolkit/ptq/pytorch/quantization_facade.py
 model_compression_toolkit/qat/__init__.py
 model_compression_toolkit/qat/common/__init__.py
 model_compression_toolkit/qat/common/constants.py
+model_compression_toolkit/qat/common/qat_config.py
 model_compression_toolkit/qat/keras/__init__.py
-model_compression_toolkit/qat/keras/qat_model_builder.py
 model_compression_toolkit/qat/keras/quantization_facade.py
 model_compression_toolkit/qat/keras/quantizer/__init__.py
-model_compression_toolkit/qat/keras/quantizer/quantization_dispatcher_builder.py
+model_compression_toolkit/qat/keras/quantizer/base_keras_qat_quantizer.py
+model_compression_toolkit/qat/keras/quantizer/quant_utils.py
+model_compression_toolkit/qat/keras/quantizer/quantization_builder.py
 model_compression_toolkit/qat/keras/quantizer/ste_rounding/__init__.py
-model_compression_toolkit/qat/keras/quantizer/ste_rounding/symmetirc_ste.py
+model_compression_toolkit/qat/keras/quantizer/ste_rounding/symmetric_ste.py
 model_compression_toolkit/qat/keras/quantizer/ste_rounding/uniform_ste.py
 model_compression_toolkit/qat/pytorch/__init__.py
 model_compression_toolkit/qat/pytorch/quantization_facade.py
-model_compression_toolkit/qunatizers_infrastructure/__init__.py
-model_compression_toolkit/qunatizers_infrastructure/common/__init__.py
-model_compression_toolkit/qunatizers_infrastructure/common/base_quantizer.py
-model_compression_toolkit/qunatizers_infrastructure/common/node_quantization_dispatcher.py
-model_compression_toolkit/qunatizers_infrastructure/keras/__init__.py
-model_compression_toolkit/qunatizers_infrastructure/keras/base_keras_quantizer.py
-model_compression_toolkit/qunatizers_infrastructure/keras/config_serialization.py
-model_compression_toolkit/qunatizers_infrastructure/keras/keras_node_quantization_dispatcher.py
-model_compression_toolkit/qunatizers_infrastructure/keras/load_model.py
-model_compression_toolkit/qunatizers_infrastructure/keras/quantize_wrapper.py
-model_compression_toolkit/qunatizers_infrastructure/pytorch/__init__.py
+model_compression_toolkit/qat/pytorch/quantizer/__init__.py
+model_compression_toolkit/qat/pytorch/quantizer/base_pytorch_qat_quantizer.py
+model_compression_toolkit/qat/pytorch/quantizer/quantization_builder.py
+model_compression_toolkit/qat/pytorch/quantizer/quantizer_utils.py
+model_compression_toolkit/qat/pytorch/quantizer/ste_rounding/__init__.py
+model_compression_toolkit/qat/pytorch/quantizer/ste_rounding/symmetric_ste.py
+model_compression_toolkit/qat/pytorch/quantizer/ste_rounding/uniform_ste.py
+model_compression_toolkit/quantizers_infrastructure/__init__.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/__init__.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/common/__init__.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/common/base_inferable_quantizer.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/common/constants.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/common/get_all_subclasses.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/common/get_quantizers.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/common/quant_utils.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/__init__.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/load_model.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantize_wrapper.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantizer_utils.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/validation_functions.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantizers/__init__.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantizers/base_keras_inferable_quantizer.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantizers/constants.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantizers/activation_inferable_quantizers/__init__.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantizers/activation_inferable_quantizers/activation_lut_pot_inferable_quantizer.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantizers/activation_inferable_quantizers/activation_pot_inferable_quantizer.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantizers/activation_inferable_quantizers/activation_symmetric_inferable_quantizer.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantizers/activation_inferable_quantizers/activation_uniform_inferable_quantizer.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantizers/weights_inferable_quantizers/__init__.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantizers/weights_inferable_quantizers/weights_lut_pot_inferable_quantizer.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantizers/weights_inferable_quantizers/weights_lut_symmetric_inferable_quantizer.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantizers/weights_inferable_quantizers/weights_pot_inferable_quantizer.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantizers/weights_inferable_quantizers/weights_symmetric_inferable_quantizer.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantizers/weights_inferable_quantizers/weights_uniform_inferable_quantizer.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/__init__.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantize_wrapper.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizer_utils.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/__init__.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/base_lut_symmetric_inferable_quantizer.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/base_pytorch_inferable_quantizer.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/base_symmetric_inferable_quantizer.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/base_uniform_inferable_quantizer.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/constants.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/activation_inferable_quantizers/__init__.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/activation_inferable_quantizers/activation_lut_pot_inferable_quantizer.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/activation_inferable_quantizers/activation_pot_inferable_quantizer.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/activation_inferable_quantizers/activation_symmetric_inferable_quantizer.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/activation_inferable_quantizers/activation_uniform_inferable_quantizer.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/weights_inferable_quantizers/__init__.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/weights_inferable_quantizers/weights_lut_pot_inferable_quantizer.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/weights_inferable_quantizers/weights_lut_symmetric_inferable_quantizer.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/weights_inferable_quantizers/weights_pot_inferable_quantizer.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/weights_inferable_quantizers/weights_symmetric_inferable_quantizer.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/weights_inferable_quantizers/weights_uniform_inferable_quantizer.py
+model_compression_toolkit/quantizers_infrastructure/trainable_infrastructure/__init__.py
+model_compression_toolkit/quantizers_infrastructure/trainable_infrastructure/common/__init__.py
+model_compression_toolkit/quantizers_infrastructure/trainable_infrastructure/common/base_trainable_quantizer.py
+model_compression_toolkit/quantizers_infrastructure/trainable_infrastructure/common/get_quantizer_config.py
+model_compression_toolkit/quantizers_infrastructure/trainable_infrastructure/common/get_quantizers.py
+model_compression_toolkit/quantizers_infrastructure/trainable_infrastructure/common/quant_utils.py
+model_compression_toolkit/quantizers_infrastructure/trainable_infrastructure/common/trainable_quantizer_config.py
+model_compression_toolkit/quantizers_infrastructure/trainable_infrastructure/keras/__init__.py
+model_compression_toolkit/quantizers_infrastructure/trainable_infrastructure/keras/base_keras_quantizer.py
+model_compression_toolkit/quantizers_infrastructure/trainable_infrastructure/keras/config_serialization.py
+model_compression_toolkit/quantizers_infrastructure/trainable_infrastructure/keras/quantizer_utils.py
+model_compression_toolkit/quantizers_infrastructure/trainable_infrastructure/pytorch/__init__.py
+model_compression_toolkit/quantizers_infrastructure/trainable_infrastructure/pytorch/base_pytorch_quantizer.py

{mct-nightly-1.7.1.31122022.post351 → mct-nightly-1.8.0.1042023.post423}/model_compression_toolkit/__init__.py RENAMED Viewed

@@ -14,8 +14,6 @@
 # ==============================================================================
 from model_compression_toolkit.core.common.quantization.debug_config import DebugConfig
-from model_compression_toolkit.gptq.common.gptq_config import GradientPTQConfig, RoundingType, GumbelConfig, \
-    GradientPTQConfigV2
 from model_compression_toolkit.core.common.quantization import quantization_config
 from model_compression_toolkit.core.common.mixed_precision import mixed_precision_quantization_config
 from model_compression_toolkit.core.common.quantization.quantization_config import QuantizationConfig, \
@@ -26,6 +24,7 @@ from model_compression_toolkit.core.tpc_models.get_target_platform_capabilities
 from model_compression_toolkit.core.common.mixed_precision.kpi_tools.kpi import KPI
 from model_compression_toolkit.core.common.mixed_precision.mixed_precision_quantization_config import \
     MixedPrecisionQuantizationConfig, MixedPrecisionQuantizationConfigV2
+from model_compression_toolkit.qat.common.qat_config import QATConfig, TrainingMethod
 from model_compression_toolkit.core.common.logger import set_log_folder
 from model_compression_toolkit.core.common.data_loader import FolderImageLoader
 from model_compression_toolkit.core.common.framework_info import FrameworkInfo, ChannelAxis
@@ -35,21 +34,21 @@ from model_compression_toolkit.core.common import network_editors as network_edi
 from model_compression_toolkit.core.keras.quantization_facade import keras_post_training_quantization, \
     keras_post_training_quantization_mixed_precision
 from model_compression_toolkit.ptq.keras.quantization_facade import keras_post_training_quantization_experimental
-from model_compression_toolkit.gptq.keras.quantization_facade import \
-    keras_gradient_post_training_quantization_experimental
-from model_compression_toolkit.gptq.keras.quantization_facade import get_keras_gptq_config
-from model_compression_toolkit.qat.keras.quantization_facade import keras_quantization_aware_training_init, \
-    keras_quantization_aware_training_finalize
-from model_compression_toolkit.core.pytorch.quantization_facade import pytorch_post_training_quantization, \
-    pytorch_post_training_quantization_mixed_precision
+from model_compression_toolkit.qat.keras.quantization_facade import keras_quantization_aware_training_init, keras_quantization_aware_training_finalize
+from model_compression_toolkit.qat.pytorch.quantization_facade import pytorch_quantization_aware_training_init, pytorch_quantization_aware_training_finalize
+from model_compression_toolkit.core.pytorch.quantization_facade import pytorch_post_training_quantization, pytorch_post_training_quantization_mixed_precision
 from model_compression_toolkit.ptq.pytorch.quantization_facade import pytorch_post_training_quantization_experimental
-from model_compression_toolkit.gptq.pytorch.quantization_facade import \
-    pytorch_gradient_post_training_quantization_experimental
-from model_compression_toolkit.gptq.pytorch.quantization_facade import get_pytorch_gptq_config
 from model_compression_toolkit.core.keras.kpi_data_facade import keras_kpi_data, keras_kpi_data_experimental
 from model_compression_toolkit.core.pytorch.kpi_data_facade import pytorch_kpi_data, pytorch_kpi_data_experimental
-from model_compression_toolkit.qunatizers_infrastructure.keras.load_model import keras_load_quantized_model
+from model_compression_toolkit.quantizers_infrastructure.inferable_infrastructure.keras.load_model import keras_load_quantized_model
-__version__ = "1.7.1"
+from model_compression_toolkit import exporter
+from model_compression_toolkit import gptq
+from model_compression_toolkit.gptq import GradientPTQConfig
+__version__ = "1.8.0"

{mct-nightly-1.7.1.31122022.post351 → mct-nightly-1.8.0.1042023.post423}/model_compression_toolkit/core/common/back2framework/base_model_builder.py RENAMED Viewed

@@ -51,4 +51,4 @@ class BaseModelBuilder(ABC):
         Returns: A framework's model built from its graph.
         """
-        raise NotImplemented(f'{self.__class__.__name__} have to implement build_model method.')
+        raise NotImplemented(f'{self.__class__.__name__} have to implement build_model method.')  # pragma: no cover

{mct-nightly-1.7.1.31122022.post351 → mct-nightly-1.8.0.1042023.post423}/model_compression_toolkit/core/common/collectors/base_collector.py RENAMED Viewed

@@ -14,6 +14,7 @@
 # ==============================================================================
 import numpy as np
+from model_compression_toolkit.core.common.logger import Logger
 class BaseCollector(object):
@@ -33,7 +34,8 @@ class BaseCollector(object):
         """
-        raise Exception(f'{self.__class__.__name__} needs to implement scale operation for its state.')
+        raise NotImplemented(
+            f'{self.__class__.__name__} needs to implement scale operation for its state.')  # pragma: no cover
     def shift(self, shift_value: np.ndarray):
         """
@@ -43,7 +45,8 @@ class BaseCollector(object):
         """
-        raise Exception(f'{self.__class__.__name__} needs to implement shift operation for its state.')
+        raise NotImplemented(
+            f'{self.__class__.__name__} needs to implement shift operation for its state.')  # pragma: no cover
     def update_legal_status(self, is_illegal: bool):
         """
@@ -63,5 +66,5 @@ class BaseCollector(object):
         """
         if not self.is_legal:
-            raise Exception(f'{self.__class__.__name__} was manipulated per-channel,'
-                            'but collected per-tensor. Data is invalid.')
+            Logger.exception(f'{self.__class__.__name__} was manipulated per-channel,'
+                             'but collected per-tensor. Data is invalid.')  # pragma: no cover

{mct-nightly-1.7.1.31122022.post351 → mct-nightly-1.8.0.1042023.post423}/model_compression_toolkit/core/common/collectors/statistics_collector.py RENAMED Viewed

@@ -37,7 +37,7 @@ class BaseStatsCollector(object):
         Returns whether this tensor requires statistics collection or not.
         Should be implemented in extending classes.
         """
-        raise Exception(f'require_collection is not implemented in {self.__class__.__name__}')
+        raise NotImplemented(f'require_collection is not implemented in {self.__class__.__name__}')  # pragma: no cover
     def update_statistics(self,
                           x: Any):
@@ -47,7 +47,7 @@ class BaseStatsCollector(object):
         Args:
             x: Tensor.
         """
-        raise Exception(f'update_statistics is not implemented in {self.__class__.__name__}')
+        raise NotImplemented(f'update_statistics is not implemented in {self.__class__.__name__}')  # pragma: no cover
 class StatsCollector(BaseStatsCollector):

{mct-nightly-1.7.1.31122022.post351 → mct-nightly-1.8.0.1042023.post423}/model_compression_toolkit/core/common/constants.py RENAMED Viewed

@@ -21,10 +21,11 @@ FOUND_TF = importlib.util.find_spec(TENSORFLOW) is not None and importlib.util.f
     "tensorflow_model_optimization") is not None
 FOUND_TORCH = importlib.util.find_spec("torch") is not None
 FOUND_ONNX = importlib.util.find_spec("onnx") is not None
+FOUND_ONNXRUNTIME = importlib.util.find_spec("onnxruntime") is not None
 WEIGHTS_SIGNED = True
 # Minimal threshold to use for quantization ranges:
-MIN_THRESHOLD = (2 ** -28)
+MIN_THRESHOLD = (2 ** -16)
 EPS = 1e-8
 MULTIPLIER_N_BITS = 8
@@ -114,12 +115,16 @@ ACTIVATION_QUANT_PARAMS_FN = 'activation_quantization_params_fn'
 WEIGHTS_QUANT_PARAMS_FN = 'weights_quantization_params_fn'
 WEIGHTS_CHANNELS_AXIS = 'weights_channels_axis'
-# GPTQ Parameters
-GUMBEL_MAX_ITER = 10000
 # Memory graph constants
 DUMMY_NODE = 'dummy_node'
 DUMMY_TENSOR = 'dummy_tensor'
 # TP Model constants
 OPS_SET_LIST = 'ops_set_list'
+# TF Input node base name
+INPUT_BASE_NAME = 'base_input'
+# Jacobian-weights constants
+MIN_JACOBIANS_ITER = 10
+JACOBIANS_COMP_TOLERANCE = 1e-3

mct-nightly 1.7.1.31122022.post351__tar.gz → 1.8.0.1042023.post423__tar.gz

mct-nightly 1.7.1.31122022.post351tar.gz → 1.8.0.1042023.post423tar.gz