Skip to content

Commit aa956bc

Browse files
authored
AC: fix batch size settings (#3427)
1 parent e9b5cfd commit aa956bc

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

tools/accuracy_checker/openvino/tools/accuracy_checker/launcher/input_feeder.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -405,7 +405,6 @@ def _parse_inputs_config(self, inputs_entry, default_layout='NCHW', precisions_l
405405
inputs_entry, precision_info, not_config_inputs, precisions, layouts_info, layouts
406406
)
407407

408-
409408
return (
410409
constant_inputs,
411410
non_constant_inputs,

tools/accuracy_checker/openvino/tools/accuracy_checker/launcher/openvino_launcher.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -574,8 +574,11 @@ def _get_model_batch_size(self):
574574
input_info = input_nodes[0]
575575
layout = (
576576
self._process_layout(input_info.get_node().layout, input_info.get_node().friendly_name)
577-
or self.default_layout
577+
or ''
578578
)
579+
input_shape = parse_partial_shape(input_info.partial_shape)
580+
if not layout and len(input_shape) == len(self.default_layout):
581+
layout = self.default_layout
579582
batch_pos = layout.find('N')
580583
if batch_pos != -1:
581584
return parse_partial_shape(input_info.partial_shape)[batch_pos]

0 commit comments

Comments
 (0)