From 27b62d9dec86b76422c6c8be7615c1f712518570 Mon Sep 17 00:00:00 2001 From: alexander Date: Tue, 4 May 2021 20:46:08 +0100 Subject: If inference fails - exit before printing results. It was easy to miss error message and mistakenly assume that execution succeeded. Signed-off-by: alexander Change-Id: I3c7beb5afe97db5df486e12e3bfe0e8e27e382f8 --- source/use_case/kws_asr/src/UseCaseHandler.cc | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) (limited to 'source/use_case/kws_asr/src/UseCaseHandler.cc') diff --git a/source/use_case/kws_asr/src/UseCaseHandler.cc b/source/use_case/kws_asr/src/UseCaseHandler.cc index 1edc7c4..7025d6d 100644 --- a/source/use_case/kws_asr/src/UseCaseHandler.cc +++ b/source/use_case/kws_asr/src/UseCaseHandler.cc @@ -238,7 +238,10 @@ namespace app { audioDataSlider.TotalStrides() + 1); /* Run inference over this audio clip sliding window. */ - arm::app::RunInference(kwsModel, profiler); + if (!RunInference(kwsModel, profiler)) { + printf_err("KWS inference failed\n"); + return output; + } std::vector kwsClassificationResult; auto& kwsClassifier = ctx.Get("kwsclassifier"); @@ -391,7 +394,10 @@ namespace app { asrPrep.Invoke(asrInferenceWindow, asrInferenceWindowLen, asrInputTensor); /* Run inference over this audio clip sliding window. */ - arm::app::RunInference(asrModel, profiler); + if (!RunInference(asrModel, profiler)) { + printf_err("ASR inference failed\n"); + return false; + } /* Post-process. */ asrPostp.Invoke(asrOutputTensor, reductionAxis, !audioDataSlider.HasNext()); -- cgit v1.2.1