mct-nightly 1.10.0.20240101.post439__py3-none-any.whl → 1.10.0.20240102.post421__py3-none-any.whl

{mct_nightly-1.10.0.20240101.post439.dist-info → mct_nightly-1.10.0.20240102.post421.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mct-nightly
-Version: 1.10.0.20240101.post439
+Version: 1.10.0.20240102.post421
 Summary: A Model Compression Toolkit for neural networks
 Home-page: UNKNOWN
 License: UNKNOWN

{mct_nightly-1.10.0.20240101.post439.dist-info → mct_nightly-1.10.0.20240102.post421.dist-info}/RECORD

@@ -303,12 +303,12 @@ model_compression_toolkit/exporter/model_exporter/fw_agonstic/__init__.py,sha256
 model_compression_toolkit/exporter/model_exporter/fw_agonstic/exporter.py,sha256=eSC6gEMc9KY5EwVRam9pJCBpCm0ksUeobKV_JAOap9M,2017
 model_compression_toolkit/exporter/model_exporter/fw_agonstic/quantization_format.py,sha256=oRlw_LG0UNi8Vl77E1U6WDBsEp1eNaphUbG9rSZkSl4,800
 model_compression_toolkit/exporter/model_exporter/keras/__init__.py,sha256=uZ2RigbY9O2PJ0Il8wPpS_s7frgg9WUGd_SHeKGyl1A,699
-model_compression_toolkit/exporter/model_exporter/keras/base_keras_exporter.py,sha256=Are89mV0cPZ11PSHnriWCMJzTLPhughxZNwdEo_aqU8,1690
-model_compression_toolkit/exporter/model_exporter/keras/export_serialization_format.py,sha256=mfc-0Mmajt4fstbwQPWGbeZYb4yexij8Fkv-Vll7sNk,1014
+model_compression_toolkit/exporter/model_exporter/keras/base_keras_exporter.py,sha256=-wr2n0yRlmFixXBeZuxg6Rzlvz-ZFUX-PJgSXhgMrEo,1593
+model_compression_toolkit/exporter/model_exporter/keras/export_serialization_format.py,sha256=v_-rOsWDFI-3k8CoJIr-XzT7ny8WXpAMteWRWtTzaeg,963
 model_compression_toolkit/exporter/model_exporter/keras/fakely_quant_keras_exporter.py,sha256=E_1IqFYAGUMOrt3U_JK1k--8D0WzWPbjZH_IRLGw_wY,11478
-model_compression_toolkit/exporter/model_exporter/keras/fakely_quant_tflite_exporter.py,sha256=qiWyiIp-WLDj1r924oeQc-M4WCi-HTk2vw0ISipy7Nk,3196
+model_compression_toolkit/exporter/model_exporter/keras/fakely_quant_tflite_exporter.py,sha256=lwwbEAmPmlU89DXmBjmDT6Yk1W6ss96LQ52oAHxvtFw,3209
 model_compression_toolkit/exporter/model_exporter/keras/int8_tflite_exporter.py,sha256=nGtpDTeH5Tdp7sjyuXsy_9TPpijDYp4nkz366DUUJ0Q,8048
-model_compression_toolkit/exporter/model_exporter/keras/keras_export_facade.py,sha256=mJdiih4jQ_pr6r5PxQE4TipqpCzsH4S3Qq7zpi-8MRs,6007
+model_compression_toolkit/exporter/model_exporter/keras/keras_export_facade.py,sha256=O-GApieS7_zLkpygnN0YvDK-HkCChwA4bSExbI5jvQ8,5998
 model_compression_toolkit/exporter/model_exporter/keras/mctq_keras_exporter.py,sha256=qXXkv3X_wb7t622EOHwXIxfGLGaDqh0T0y4UxREi4Bo,1976
 model_compression_toolkit/exporter/model_exporter/pytorch/__init__.py,sha256=uZ2RigbY9O2PJ0Il8wPpS_s7frgg9WUGd_SHeKGyl1A,699
 model_compression_toolkit/exporter/model_exporter/pytorch/base_pytorch_exporter.py,sha256=UPVkEUQCMZ4Lld6CRnEOPEmlfe5vcQZG0Q3FwRBodD4,4021
@@ -474,8 +474,8 @@ model_compression_toolkit/trainable_infrastructure/keras/quantize_wrapper.py,sha
 model_compression_toolkit/trainable_infrastructure/keras/quantizer_utils.py,sha256=MVwXNymmFRB2NXIBx4e2mdJ1RfoHxRPYRgjb1MQP5kY,1797
 model_compression_toolkit/trainable_infrastructure/pytorch/__init__.py,sha256=huHoBUcKNB6BnY6YaUCcFvdyBtBI172ZoUD8ZYeNc6o,696
 model_compression_toolkit/trainable_infrastructure/pytorch/base_pytorch_quantizer.py,sha256=SbvRlIdE32PEBsINt1bhSqvrKL_zbM9V-aeSkOn-sw4,3083
-mct_nightly-1.10.0.20240101.post439.dist-info/LICENSE.md,sha256=aYSSIb-5AFPeITTvXm1UAoe0uYBiMmSS8flvXaaFUks,10174
-mct_nightly-1.10.0.20240101.post439.dist-info/METADATA,sha256=JJl4aiW3kYdqZ4N8m_Fn-fPvOPHjS4AP6Dd4vjS8gCM,17141
-mct_nightly-1.10.0.20240101.post439.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-mct_nightly-1.10.0.20240101.post439.dist-info/top_level.txt,sha256=gsYA8juk0Z-ZmQRKULkb3JLGdOdz8jW_cMRjisn9ga4,26
-mct_nightly-1.10.0.20240101.post439.dist-info/RECORD,,
+mct_nightly-1.10.0.20240102.post421.dist-info/LICENSE.md,sha256=aYSSIb-5AFPeITTvXm1UAoe0uYBiMmSS8flvXaaFUks,10174
+mct_nightly-1.10.0.20240102.post421.dist-info/METADATA,sha256=Wo83EPwubPoSUM-v8xvhW2zuWl0M4cOeXNlnEwtlvyw,17141
+mct_nightly-1.10.0.20240102.post421.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+mct_nightly-1.10.0.20240102.post421.dist-info/top_level.txt,sha256=gsYA8juk0Z-ZmQRKULkb3JLGdOdz8jW_cMRjisn9ga4,26
+mct_nightly-1.10.0.20240102.post421.dist-info/RECORD,,

model_compression_toolkit/exporter/model_exporter/keras/base_keras_exporter.py

@@ -19,10 +19,8 @@ import keras
 
 import tensorflow as tf
 from packaging import version
-if version.parse(tf.__version__) >= version.parse("2.13"):
-    TMP_KERAS_EXPORT_FORMAT = ".keras"
-else:
-    TMP_KERAS_EXPORT_FORMAT = ".h5"
+
+DEFAULT_KERAS_EXPORT_EXTENTION = '.keras'
 
 
 class BaseKerasExporter(Exporter):

model_compression_toolkit/exporter/model_exporter/keras/export_serialization_format.py

@@ -19,11 +19,11 @@ class KerasExportSerializationFormat(Enum):
     """
     Specify which serialization format to use for exporting a quantized Keras model.
 
-    KERAS_H5 - .keras (TF2.13 and above) or .h5 (TF2.12 and below) file format
+    KERAS - .keras file format
 
     TFLITE - .tflite file format
 
     """
 
-    KERAS_H5 = 0
+    KERAS = 0
     TFLITE = 1

model_compression_toolkit/exporter/model_exporter/keras/fakely_quant_tflite_exporter.py

@@ -20,10 +20,9 @@ import tensorflow as tf
 import keras.models
 
 from model_compression_toolkit.logger import Logger
-from model_compression_toolkit.exporter.model_exporter.keras.base_keras_exporter import TMP_KERAS_EXPORT_FORMAT
 from model_compression_toolkit.exporter.model_exporter.keras.fakely_quant_keras_exporter import FakelyQuantKerasExporter
 from model_compression_toolkit.trainable_infrastructure.keras.load_model import keras_load_quantized_model
-
+from model_compression_toolkit.exporter.model_exporter.keras.base_keras_exporter import DEFAULT_KERAS_EXPORT_EXTENTION
 
 class FakelyQuantTFLiteExporter(FakelyQuantKerasExporter):
     """
@@ -58,7 +57,7 @@ class FakelyQuantTFLiteExporter(FakelyQuantKerasExporter):
         """
         # Use Keras exporter to quantize model's weights before converting it to TFLite.
         # Since exporter saves the model, we use a tmp path for saving, and then we delete it automatically.
-        with tempfile.NamedTemporaryFile(suffix=TMP_KERAS_EXPORT_FORMAT) as tmp_file:
+        with tempfile.NamedTemporaryFile(suffix=DEFAULT_KERAS_EXPORT_EXTENTION) as tmp_file:
             FakelyQuantKerasExporter(self.model,
                                      self.is_layer_exportable_fn,
                                      tmp_file.name,

model_compression_toolkit/exporter/model_exporter/keras/keras_export_facade.py

@@ -32,14 +32,14 @@ if FOUND_TF:
     from model_compression_toolkit.exporter.model_exporter.keras.mctq_keras_exporter import MCTQKerasExporter
 
     supported_serialization_quantization_export_dict = {
-        KerasExportSerializationFormat.KERAS_H5: [QuantizationFormat.FAKELY_QUANT, QuantizationFormat.MCTQ],
+        KerasExportSerializationFormat.KERAS: [QuantizationFormat.FAKELY_QUANT, QuantizationFormat.MCTQ],
         KerasExportSerializationFormat.TFLITE: [QuantizationFormat.FAKELY_QUANT, QuantizationFormat.INT8]
     }
 
     def keras_export_model(model: keras.models.Model,
                            save_model_path: str,
                            is_layer_exportable_fn: Callable = is_keras_layer_exportable,
-                           serialization_format: KerasExportSerializationFormat = KerasExportSerializationFormat.KERAS_H5,
+                           serialization_format: KerasExportSerializationFormat = KerasExportSerializationFormat.KERAS,
                            quantization_format : QuantizationFormat = QuantizationFormat.MCTQ) -> Dict[str, type]:
         """
         Export a Keras quantized model to a h5 or tflite model.
@@ -63,7 +63,7 @@ if FOUND_TF:
 
         """
 
-        if serialization_format == KerasExportSerializationFormat.KERAS_H5:
+        if serialization_format == KerasExportSerializationFormat.KERAS:
             if quantization_format == QuantizationFormat.FAKELY_QUANT:
                 exporter = FakelyQuantKerasExporter(model,
                                                     is_layer_exportable_fn,