saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Allow for optional graph inputs in InferenceSession.run

Browse source
This commit is contained in:
Scott McKay 2018-11-22 11:53:14 +10:00
parent ba470c2180
commit 0ee3ad32d2
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
onnxruntime/python/session.py
View file

@ -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]