fix metric again

sony · Mar 10, 2024 · 78c0b9e · 78c0b9e
1 parent 008918a
commit 78c0b9e
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/model_compression_toolkit/core/common/mixed_precision/search_methods/linear_programming.py b/model_compression_toolkit/core/common/mixed_precision/search_methods/linear_programming.py
@@ -299,8 +299,7 @@ def _build_layer_to_metrics_mapping(search_manager: MixedPrecisionSearchManager,
                     [node_idx],
                     search_manager.max_kpi_config)
 
-            layer_to_metrics_mapping[node_idx][bitwidth_idx] = max(metric_value,
-                                                                   layer_to_metrics_mapping[node_idx][bitwidth_idx-1] + eps)
+            layer_to_metrics_mapping[node_idx][bitwidth_idx] = max(metric_value, max_config_value + eps)
 
     # Finalize distance metric mapping
     search_manager.finalize_distance_metric(layer_to_metrics_mapping)