- Fix the retrieval of input shape from the openvino network (#46)

nebuly-ai · May 5, 2022 · f656461 · f656461
1 parent fc778b6
commit f656461
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nebullvm/inference_learners/openvino.py b/nebullvm/inference_learners/openvino.py
@@ -140,7 +140,7 @@ def from_model_name(
     def _rebuild_network(self, input_shapes: Dict):
         network = self.exec_network.get_exec_graph_info()
         if all(
-            input_shape == tuple(network.inputs[input_name].shape)
+            input_shape == tuple(network.input_info[input_name].input_data.shape)
             for input_name, input_shape in input_shapes.items()
         ):
             # If the new input shapes is equal to the previous one do nothing.