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

Merge pull request #17 from Microsoft/scmckay/AllowForOptionalInputsInSessionPy

Browse source 
Allow for optional graph inputs in InferenceSession.run
This commit is contained in:
Pranav Sharma 2018-11-22 22:20:19 -08:00 committed by GitHub
commit faa06d3516
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
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]