mct-nightly 2.1.0.20240616.434__py3-none-any.whl → 2.1.0.20240616.65727__py3-none-any.whl

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 (7) hide show
  1. {mct_nightly-2.1.0.20240616.434.dist-info → mct_nightly-2.1.0.20240616.65727.dist-info}/METADATA +1 -1
  2. {mct_nightly-2.1.0.20240616.434.dist-info → mct_nightly-2.1.0.20240616.65727.dist-info}/RECORD +7 -7
  3. model_compression_toolkit/__init__.py +1 -1
  4. model_compression_toolkit/core/keras/hessian/weights_trace_hessian_calculator_keras.py +1 -1
  5. {mct_nightly-2.1.0.20240616.434.dist-info → mct_nightly-2.1.0.20240616.65727.dist-info}/LICENSE.md +0 -0
  6. {mct_nightly-2.1.0.20240616.434.dist-info → mct_nightly-2.1.0.20240616.65727.dist-info}/WHEEL +0 -0
  7. {mct_nightly-2.1.0.20240616.434.dist-info → mct_nightly-2.1.0.20240616.65727.dist-info}/top_level.txt +0 -0
{mct_nightly-2.1.0.20240616.434.dist-info → mct_nightly-2.1.0.20240616.65727.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: mct-nightly
3
- Version: 2.1.0.20240616.434
3
+ Version: 2.1.0.20240616.65727
4
4
  Summary: A Model Compression Toolkit for neural networks
5
5
  Home-page: UNKNOWN
6
6
  License: UNKNOWN
{mct_nightly-2.1.0.20240616.434.dist-info → mct_nightly-2.1.0.20240616.65727.dist-info}/RECORD RENAMED
@@ -1,4 +1,4 @@
1
- model_compression_toolkit/__init__.py,sha256=mregiWggl4blOSBp61pEozF4QXuJnyAsxzpvrXIwh2k,1573
1
+ model_compression_toolkit/__init__.py,sha256=tiJgaUR3HmSVpDI-Vv2wT3GOa8-4cZ_SfpIM_6GpwV4,1573
2
2
  model_compression_toolkit/constants.py,sha256=9pVleMwnhlM4QwIL2HcEq42I1uF4rlSw63RUjkxOF4w,3923
3
3
  model_compression_toolkit/defaultdict.py,sha256=LSc-sbZYXENMCw3U9F4GiXuv67IKpdn0Qm7Fr11jy-4,2277
4
4
  model_compression_toolkit/logger.py,sha256=3DByV41XHRR3kLTJNbpaMmikL8icd9e1N-nkQAY9oDk,4567
@@ -186,7 +186,7 @@ model_compression_toolkit/core/keras/graph_substitutions/substitutions/weights_a
186
186
  model_compression_toolkit/core/keras/hessian/__init__.py,sha256=lNJ29DYxaLUPDstRDA1PGI5r9Fulq_hvrZMlhst1Z5g,697
187
187
  model_compression_toolkit/core/keras/hessian/activation_trace_hessian_calculator_keras.py,sha256=4eJKq_Fx4mm_VuBDeeti0fTcUk1lL2yjebxCugJhvrA,8871
188
188
  model_compression_toolkit/core/keras/hessian/trace_hessian_calculator_keras.py,sha256=hRfAjgZakDaIMuERmTVjJSa_Ww6FmEudYPO9R7SuYuQ,3914
189
- model_compression_toolkit/core/keras/hessian/weights_trace_hessian_calculator_keras.py,sha256=KBjGr9FzyZIPD4MFtsV3LDBdJtLa0VFdIXyx_KAnjTQ,12215
189
+ model_compression_toolkit/core/keras/hessian/weights_trace_hessian_calculator_keras.py,sha256=Xogd90kPZPvKbplZQv5B77Dq_m4aW5-bL6Jxh33VZWs,12213
190
190
  model_compression_toolkit/core/keras/mixed_precision/__init__.py,sha256=sw7LOPN1bM82o3SkMaklyH0jw-TLGK0-fl2Wq73rffI,697
191
191
  model_compression_toolkit/core/keras/mixed_precision/configurable_activation_quantizer.py,sha256=aW8wR13fK6P6xzbU9XGU60IO1yYzXSo_Hk4qeq486kg,5137
192
192
  model_compression_toolkit/core/keras/mixed_precision/configurable_weights_quantizer.py,sha256=Ziydik2j-LvNBXP3TSfUD6rEezPAikzQGib0_IXkmGM,6729
@@ -491,8 +491,8 @@ model_compression_toolkit/trainable_infrastructure/keras/quantize_wrapper.py,sha
491
491
  model_compression_toolkit/trainable_infrastructure/keras/quantizer_utils.py,sha256=MVwXNymmFRB2NXIBx4e2mdJ1RfoHxRPYRgjb1MQP5kY,1797
492
492
  model_compression_toolkit/trainable_infrastructure/pytorch/__init__.py,sha256=huHoBUcKNB6BnY6YaUCcFvdyBtBI172ZoUD8ZYeNc6o,696
493
493
  model_compression_toolkit/trainable_infrastructure/pytorch/base_pytorch_quantizer.py,sha256=MxylaVFPgN7zBiRBy6WV610EA4scLgRJFbMucKvvNDU,2896
494
- mct_nightly-2.1.0.20240616.434.dist-info/LICENSE.md,sha256=aYSSIb-5AFPeITTvXm1UAoe0uYBiMmSS8flvXaaFUks,10174
495
- mct_nightly-2.1.0.20240616.434.dist-info/METADATA,sha256=qRPmiufe7bR0fBvN5cB16-cYIewTIbYTbbpfBQvjA_8,19721
496
- mct_nightly-2.1.0.20240616.434.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
497
- mct_nightly-2.1.0.20240616.434.dist-info/top_level.txt,sha256=gsYA8juk0Z-ZmQRKULkb3JLGdOdz8jW_cMRjisn9ga4,26
498
- mct_nightly-2.1.0.20240616.434.dist-info/RECORD,,
494
+ mct_nightly-2.1.0.20240616.65727.dist-info/LICENSE.md,sha256=aYSSIb-5AFPeITTvXm1UAoe0uYBiMmSS8flvXaaFUks,10174
495
+ mct_nightly-2.1.0.20240616.65727.dist-info/METADATA,sha256=PKQDHwJ9J3IdkJTncdUeqbFVeZ5p97EE1uJIFzb8I-E,19723
496
+ mct_nightly-2.1.0.20240616.65727.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
497
+ mct_nightly-2.1.0.20240616.65727.dist-info/top_level.txt,sha256=gsYA8juk0Z-ZmQRKULkb3JLGdOdz8jW_cMRjisn9ga4,26
498
+ mct_nightly-2.1.0.20240616.65727.dist-info/RECORD,,
model_compression_toolkit/__init__.py CHANGED
@@ -27,4 +27,4 @@ from model_compression_toolkit import data_generation
27
27
  from model_compression_toolkit import pruning
28
28
  from model_compression_toolkit.trainable_infrastructure.keras.load_model import keras_load_quantized_model
29
29
 
30
- __version__ = "2.1.0.20240616.000434"
30
+ __version__ = "2.1.0.20240616.065727"
model_compression_toolkit/core/keras/hessian/weights_trace_hessian_calculator_keras.py CHANGED
@@ -74,7 +74,7 @@ class WeightsTraceHessianCalculatorKeras(TraceHessianCalculatorKeras):
74
74
  model, _ = FloatKerasModelBuilder(graph=self.graph).build_model()
75
75
 
76
76
  # Initiate a gradient tape for automatic differentiation
77
- with (tf.GradientTape(persistent=True) as tape):
77
+ with tf.GradientTape(persistent=True) as tape:
78
78
  # Perform a forward pass (inference) to get the output, while watching
79
79
  # the input tensor for gradient computation
80
80
  tape.watch(self.input_images)