EvalStep called with wrong inputs onnxruntime_training_cxx_inline.h (#17331)

This commit is contained in:
kushalpatil07 2023-08-30 02:44:35 +05:30 committed by GitHub
parent e5ca3f3dcb
commit 7b92057376
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -68,7 +68,7 @@ inline std::vector<Value> TrainingSession::EvalStep(const std::vector<Value>& 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;
}