From 7b920573760ff8a61bdbde01d1b965e895530bb1 Mon Sep 17 00:00:00 2001 From: kushalpatil07 <44136439+kushalpatil07@users.noreply.github.com> Date: Wed, 30 Aug 2023 02:44:35 +0530 Subject: [PATCH] EvalStep called with wrong inputs onnxruntime_training_cxx_inline.h (#17331) --- .../training_api/include/onnxruntime_training_cxx_inline.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/orttraining/orttraining/training_api/include/onnxruntime_training_cxx_inline.h b/orttraining/orttraining/training_api/include/onnxruntime_training_cxx_inline.h index 0661477088..c0048458dd 100644 --- a/orttraining/orttraining/training_api/include/onnxruntime_training_cxx_inline.h +++ b/orttraining/orttraining/training_api/include/onnxruntime_training_cxx_inline.h @@ -68,7 +68,7 @@ inline std::vector TrainingSession::EvalStep(const std::vector& in RunOptions run_options; ThrowOnError(GetTrainingApi().EvalStep( p_, run_options, input_values.size(), ort_input_values, - training_model_output_count_, ort_output_values)); + eval_model_output_count_, ort_output_values)); return output_values; }