From 152ab72e06b33e69e6f5aa19cc769db5b3d260f1 Mon Sep 17 00:00:00 2001
From: NerdyLee <48589307+ljwoo94@users.noreply.github.com>
Date: Tue, 10 Sep 2024 18:15:19 +0900
Subject: [PATCH] [onert-micro] Fix TrainingDriverHandler (#13961)

This commit fixes bug in TrainingDriverHandler.
ONE-DCO-1.0-Signed-off-by: Jungwoo Lee <jade.jw.lee@samsung.com>

Co-authored-by: Jungwoo Lee <jade.jw.lee@samsung.com>
---
 .../src/TrainingDriverHandler.cpp                    | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/onert-micro/training-configure-tool/src/TrainingDriverHandler.cpp b/onert-micro/training-configure-tool/src/TrainingDriverHandler.cpp
index fe4a3ad8431..eea65336980 100644
--- a/onert-micro/training-configure-tool/src/TrainingDriverHandler.cpp
+++ b/onert-micro/training-configure-tool/src/TrainingDriverHandler.cpp
@@ -156,7 +156,7 @@ OMStatus training_configure_tool::runTrainProcessWithCurConfig(
         case onert_micro::CROSS_ENTROPY_METRICS:
         {
           float cross_entropy_metric = 0.f;
-          train_interpreter.evaluateMetric(onert_micro::CROSS_ENTROPY_METRICS,
+          train_interpreter.evaluateMetric(config, onert_micro::CROSS_ENTROPY_METRICS,
                                            reinterpret_cast<void *>(&cross_entropy_metric),
                                            cur_batch_size);
           cross_entropy_v.push_back(cross_entropy_metric);
@@ -165,7 +165,7 @@ OMStatus training_configure_tool::runTrainProcessWithCurConfig(
         case onert_micro::ACCURACY:
         {
           float accuracy = 0.f;
-          train_interpreter.evaluateMetric(onert_micro::ACCURACY,
+          train_interpreter.evaluateMetric(config, onert_micro::ACCURACY,
                                            reinterpret_cast<void *>(&accuracy), cur_batch_size);
           accuracy_v.push_back(accuracy);
         }
@@ -173,16 +173,16 @@ OMStatus training_configure_tool::runTrainProcessWithCurConfig(
         case onert_micro::MSE_METRICS:
         {
           float mse = 0.f;
-          train_interpreter.evaluateMetric(onert_micro::MSE_METRICS, reinterpret_cast<void *>(&mse),
-                                           cur_batch_size);
+          train_interpreter.evaluateMetric(config, onert_micro::MSE_METRICS,
+                                           reinterpret_cast<void *>(&mse), cur_batch_size);
           mse_v.push_back(mse);
         }
         break;
         case onert_micro::MAE_METRICS:
         {
           float mae = 0.f;
-          train_interpreter.evaluateMetric(onert_micro::MAE_METRICS, reinterpret_cast<void *>(&mae),
-                                           cur_batch_size);
+          train_interpreter.evaluateMetric(config, onert_micro::MAE_METRICS,
+                                           reinterpret_cast<void *>(&mae), cur_batch_size);
           mae_v.push_back(mae);
         }
         break;