[onert-micro] Fix TrainingDriverHandler (#13961)

This commit fixes bug in TrainingDriverHandler. ONE-DCO-1.0-Signed-off-by: Jungwoo Lee <[email protected]> Co-authored-by: Jungwoo Lee <[email protected]>
Samsung · Sep 10, 2024 · 152ab72 · 152ab72
1 parent 19edda3
commit 152ab72
Showing 1 changed file with 6 additions and 6 deletions.
diff --git 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,24 +165,24 @@ 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);
         }
         break;
         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;