Skip to content

Commit e9d2415

Browse files
[BC] Fix collected inputs (#3021)
### Changes - Fix inputs collection for specific cases. ### Reason for changes - Algorithm reliability. ### Related tickets - 148633 ### Tests - Manual
1 parent 14158ce commit e9d2415

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

nncf/quantization/algorithms/bias_correction/algorithm.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -591,10 +591,11 @@ def get_statistic_points(self, model: TModel, graph: NNCFGraph) -> StatisticPoin
591591
input_id = (biased_after_input_node.node_name, edge.input_port_id)
592592
output_id = (edge.from_node.node_name, edge.output_port_id)
593593

594+
self._collected_stat_inputs_map[input_id] = output_id
595+
594596
if edge.from_node.node_name in statistic_container:
595597
continue
596598

597-
self._collected_stat_inputs_map[input_id] = output_id
598599
statistic_point = self._backend_entity.target_point(
599600
TargetType.POST_LAYER_OPERATION, edge.from_node.node_name, port_id=edge.output_port_id
600601
)

0 commit comments

Comments
 (0)