Skip to content

Commit a932eba

Browse files
authored
AC: fix set batch size (#3149)
1 parent 0814bc3 commit a932eba

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -561,9 +561,9 @@ def _set_batch_size(self, batch_size):
561561
for input_node in self.network.inputs:
562562
layer_name = input_node.get_node().friendly_name
563563
if layer_name in self.const_inputs:
564-
input_shapes[layer_name] = parse_partial_shape(layer_name.partial_shape)
564+
input_shapes[layer_name] = parse_partial_shape(input_node.get_node().partial_shape)
565565
else:
566-
layer_shape = parse_partial_shape(layer_name.partial_shape)
566+
layer_shape = parse_partial_shape(input_node.get_node().partial_shape)
567567
layout = self.inputs[layer_name].layout
568568
if '...' in str(layout):
569569
layout = self.get_layout_from_config(layer_name)

0 commit comments

Comments
 (0)