Update quantize_model.py

Dump actual_subset_size to ov.Model (Issue openvinotoolkit#2562)
AiGaf1 · Mar 22, 2024 · 989fd95 · 989fd95
1 parent b7ba5ad
commit 989fd95
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/nncf/openvino/quantization/quantize_model.py b/nncf/openvino/quantization/quantize_model.py
@@ -96,13 +96,14 @@ def native_quantize_if_op_impl(
 
     if is_weight_compression_needed(advanced_parameters):
         compress_quantize_weights_transformation(quantized_model)
-
+    
     dump_parameters(
         quantized_model,
         {
             "preset": preset,
             "target_device": target_device.value,
             "subset_size": subset_size,
+            "actual_subset_size": min(subset_size, calibration_dataset.get_length()),
             "fast_bias_correction": fast_bias_correction,
             "model_type": model_type,
             "ignored_scope": ignored_scope,