mct-nightly 2.3.0.20250128.506__py3-none-any.whl → 2.3.0.20250129.508__py3-none-any.whl

{mct_nightly-2.3.0.20250128.506.dist-info → mct_nightly-2.3.0.20250129.508.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: mct-nightly
-Version: 2.3.0.20250128.506
+Version: 2.3.0.20250129.508
 Summary: A Model Compression Toolkit for neural networks
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: Apache Software License
@@ -13,7 +13,6 @@ Requires-Dist: networkx!=2.8.1
 Requires-Dist: tqdm
 Requires-Dist: Pillow
 Requires-Dist: numpy<2.0
-Requires-Dist: opencv-python
 Requires-Dist: scikit-image
 Requires-Dist: scikit-learn
 Requires-Dist: tensorboard

{mct_nightly-2.3.0.20250128.506.dist-info → mct_nightly-2.3.0.20250129.508.dist-info}/RECORD

@@ -1,7 +1,7 @@
-model_compression_toolkit/__init__.py,sha256=NiBWK1bu8jB9ldArr4OLTfKi6hZbuZki0TNDM_q61q4,1557
+model_compression_toolkit/__init__.py,sha256=5yoBYX4rzT_uWukrdVc6UJm2mlcKu_6gh6agaWOa4-s,1557
 model_compression_toolkit/constants.py,sha256=i_R6uXBfO1ph_X6DNJych2x59SUojfJbn7dNjs_mZnc,3846
 model_compression_toolkit/defaultdict.py,sha256=LSc-sbZYXENMCw3U9F4GiXuv67IKpdn0Qm7Fr11jy-4,2277
-model_compression_toolkit/logger.py,sha256=3DByV41XHRR3kLTJNbpaMmikL8icd9e1N-nkQAY9oDk,4567
+model_compression_toolkit/logger.py,sha256=L3q7tn3Uht0i_7phnlOWMR2Te2zvzrt2HOz9vYEInts,4529
 model_compression_toolkit/metadata.py,sha256=x_Bk4VpzILdsFax6--CZ3X18qUTP28sbF_AhoQW8dNc,4003
 model_compression_toolkit/verify_packages.py,sha256=TlS-K1EP-QsghqWUW7SDPkAJiUf7ryw4tvhFDe6rCUk,1405
 model_compression_toolkit/core/__init__.py,sha256=8a0wUNBKwTdJGDk_Ho6WQAXjGuCqQZG1FUxxJlAV8L8,2096
@@ -312,7 +312,7 @@ model_compression_toolkit/data_generation/pytorch/pytorch_data_generation.py,sha
 model_compression_toolkit/data_generation/pytorch/optimization_functions/__init__.py,sha256=huHoBUcKNB6BnY6YaUCcFvdyBtBI172ZoUD8ZYeNc6o,696
 model_compression_toolkit/data_generation/pytorch/optimization_functions/batchnorm_alignment_functions.py,sha256=dMc4zz9XfYfAT4Cxns57VgvGZWPAMfaGlWLFyCyl8TA,1968
 model_compression_toolkit/data_generation/pytorch/optimization_functions/bn_layer_weighting_functions.py,sha256=We0fVMQ4oU7Y0IWQ8fKy8KpqkIiLyKoQeF9XKAQ6TH0,3317
-model_compression_toolkit/data_generation/pytorch/optimization_functions/image_initilization.py,sha256=hhWSZ7w45dE5SQ6jM27cBkCSJObWkALs_RpD6afPi68,4753
+model_compression_toolkit/data_generation/pytorch/optimization_functions/image_initilization.py,sha256=0mV2BuegNvL9MnDBu2NiJo--4KCcdDDzbWUMU4uld5w,4678
 model_compression_toolkit/data_generation/pytorch/optimization_functions/lr_scheduler.py,sha256=NydGxFIclmrfU3HWYUrRbprg4hPt470QP6MTOMLEhRs,9172
 model_compression_toolkit/data_generation/pytorch/optimization_functions/output_loss_functions.py,sha256=PRVmn8o2hTdwTdbd2ezf__LNbFvcgiVO0c25dsyg3Tg,6549
 model_compression_toolkit/data_generation/pytorch/optimization_functions/scheduler_step_functions.py,sha256=zMjY2y4FSHonuY5hddbMTb8qAQtLtohYF7q1wuruDDs,3267
@@ -523,8 +523,8 @@ model_compression_toolkit/xquant/pytorch/model_analyzer.py,sha256=b93o800yVB3Z-i
 model_compression_toolkit/xquant/pytorch/pytorch_report_utils.py,sha256=UVN_S9ULHBEldBpShCOt8-soT8YTQ5oE362y96qF_FA,3950
 model_compression_toolkit/xquant/pytorch/similarity_functions.py,sha256=CERxq5K8rqaiE-DlwhZBTUd9x69dtYJlkHOPLB54vm8,2354
 model_compression_toolkit/xquant/pytorch/tensorboard_utils.py,sha256=mkoEktLFFHtEKzzFRn_jCnxjhJolK12TZ5AQeDHzUO8,9767
-mct_nightly-2.3.0.20250128.506.dist-info/LICENSE.md,sha256=aYSSIb-5AFPeITTvXm1UAoe0uYBiMmSS8flvXaaFUks,10174
-mct_nightly-2.3.0.20250128.506.dist-info/METADATA,sha256=XOxQZqPV2WdCEKPS9RStWAOlaz2lq2qpLuo9ZyQb5CU,26601
-mct_nightly-2.3.0.20250128.506.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-mct_nightly-2.3.0.20250128.506.dist-info/top_level.txt,sha256=gsYA8juk0Z-ZmQRKULkb3JLGdOdz8jW_cMRjisn9ga4,26
-mct_nightly-2.3.0.20250128.506.dist-info/RECORD,,
+mct_nightly-2.3.0.20250129.508.dist-info/LICENSE.md,sha256=aYSSIb-5AFPeITTvXm1UAoe0uYBiMmSS8flvXaaFUks,10174
+mct_nightly-2.3.0.20250129.508.dist-info/METADATA,sha256=V1ZMks36vbn2kcBLkb88KKI-viLM_xXXWWIPogCNTnI,26572
+mct_nightly-2.3.0.20250129.508.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+mct_nightly-2.3.0.20250129.508.dist-info/top_level.txt,sha256=gsYA8juk0Z-ZmQRKULkb3JLGdOdz8jW_cMRjisn9ga4,26
+mct_nightly-2.3.0.20250129.508.dist-info/RECORD,,

model_compression_toolkit/__init__.py

@@ -27,4 +27,4 @@ from model_compression_toolkit import data_generation
 from model_compression_toolkit import pruning
 from model_compression_toolkit.trainable_infrastructure.keras.load_model import keras_load_quantized_model
 
-__version__ = "2.3.0.20250128.000506"
+__version__ = "2.3.0.20250129.000508"

model_compression_toolkit/data_generation/pytorch/optimization_functions/image_initilization.py

@@ -15,11 +15,10 @@
 from functools import partial
 from typing import Tuple, Union, List, Callable, Dict
 
-import cv2
 from torch import Tensor
 from torchvision.transforms.transforms import _setup_size
 import torch
-import numpy as np
+import torch.nn.functional as F
 from torch.utils.data import Dataset, DataLoader
 
 from model_compression_toolkit.data_generation.common.enums import DataInitType
@@ -97,9 +96,8 @@ def diverse_sample(size: Tuple[int, ...]) -> Tensor:
     sample = random_std * torch.randn(size) + random_mean
 
     # filtering to make the image a bit smoother
-    kernel = np.ones((5, 5), np.float32) / 16
-    if sample.shape[1] < 500 and sample.shape[2] < 500:
-        sample = torch.from_numpy(cv2.filter2D(sample.float().detach().cpu().numpy(), -1, kernel))
+    kernel = torch.ones(NUM_INPUT_CHANNELS, NUM_INPUT_CHANNELS, 5, 5) / 16
+    sample = F.conv2d(sample, kernel, padding=1)
     return sample.float()
 
 def default_data_init_fn(

model_compression_toolkit/logger.py

@@ -136,7 +136,6 @@ class Logger:
             msg: Message to log.
 
         """
-        print(msg)
         Logger.get_logger().info(msg)
 
     @staticmethod
@@ -148,7 +147,6 @@ class Logger:
             msg: Message to log.
 
         """
-        print(msg)
         Logger.get_logger().warning(msg)
 
     @staticmethod