Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Don't try to bind unused inputs in the Training frontend #6166

Merged
merged 11 commits into from
Dec 18, 2020
Prev Previous commit
Next Next commit
Add temp debug prints.
  • Loading branch information
Sergii Dymchenko committed Dec 18, 2020
commit c90ee11e53a129556d5c56e165efcfd70b4992bb
2 changes: 1 addition & 1 deletion orttraining/orttraining/python/training/orttrainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -804,7 +804,7 @@ def _training_session_run_helper(self, is_train, inputs, inputs_desc, outputs_de
else:
iobinding = self._eval_io_binding

print(self._training_session.get_inputs(self))
print("x"*10, ": ", self._training_session.get_inputs())
# Bind input tensors
for input, input_desc in zip(inputs, inputs_desc):
device_index = _utils.get_device_index_from_input(input)
Expand Down