From 0ee3ad32d2043f171aead47ad1384d039a4ecba6 Mon Sep 17 00:00:00 2001 From: Scott McKay Date: Thu, 22 Nov 2018 11:53:14 +1000 Subject: [PATCH] Allow for optional graph inputs in InferenceSession.run --- onnxruntime/python/session.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/onnxruntime/python/session.py b/onnxruntime/python/session.py index a97b8685fa..584473d591 100644 --- a/onnxruntime/python/session.py +++ b/onnxruntime/python/session.py @@ -64,7 +64,8 @@ class InferenceSession: """ num_required_inputs = len(self._inputs_meta) num_inputs = len(input_feed) - if num_inputs != num_required_inputs: + # the graph may have optional inputs used to override initializers. allow for that. + if num_inputs < num_required_inputs: raise ValueError("Model requires {} inputs. Input Feed contains {}".format(num_required_inputs, num_inputs)) if not output_names: output_names = [output.name for output in self._outputs_meta]