From 3270bf57c194d0406ac563de8790e288539df2eb Mon Sep 17 00:00:00 2001 From: Seungbaek Hong Date: Tue, 4 Apr 2023 15:45:49 +0900 Subject: [PATCH] [Ahub] Fix svace issue Fixed some svace issues on svace-tizen_7.5. Add exception handling to - tensor_trainer_nntrainer.cc - genModelExeOrder.cpp Signed-off-by: Seungbaek Hong --- nnstreamer/tensor_trainer/tensor_trainer_nntrainer.cc | 5 +++++ test/input_gen/genModelExeOrder.cpp | 8 +++++++- 2 files changed, 12 insertions(+), 1 deletion(-) diff --git a/nnstreamer/tensor_trainer/tensor_trainer_nntrainer.cc b/nnstreamer/tensor_trainer/tensor_trainer_nntrainer.cc index f1d3e633..67bcec3f 100644 --- a/nnstreamer/tensor_trainer/tensor_trainer_nntrainer.cc +++ b/nnstreamer/tensor_trainer/tensor_trainer_nntrainer.cc @@ -160,6 +160,9 @@ static int nntrainer_model_push_data(const GstTensorTrainerFramework *fw, nntrainer->num_validation_samples) { data = nntrainer->valid_data.get(); ml_logd("#### V-Data ####"); + } else { + ml_loge("Invalid push_count"); + return -1; } ml_logd("number of inputs(%" PRId64 ") and labels(%" PRId64 ")", @@ -531,6 +534,8 @@ void NNTrainer::NNTrainerTrain::createModel() { NNTrainer::NNTrainerTrain::NNTrainerTrain( const GstTensorTrainerProperties *prop, const std::string &_model_config) : + training_loss(0), + validation_loss(0), num_push_data(0), model_config(_model_config) { ml_logd(""); diff --git a/test/input_gen/genModelExeOrder.cpp b/test/input_gen/genModelExeOrder.cpp index bae8f770..5f78624a 100644 --- a/test/input_gen/genModelExeOrder.cpp +++ b/test/input_gen/genModelExeOrder.cpp @@ -114,7 +114,13 @@ std::vector> test_models = { int main(int argc, char **argv) { for (auto &[name, model] : test_models) { - exportToFile(name, model); + try { + exportToFile(name, model); + } catch (const std::exception &e) { + ml_loge("Got error while export file. %s, %s", typeid(e).name(), + e.what()); + return EXIT_FAILURE; + } } return EXIT_SUCCESS; -- 2.34.1