rebase fixes

Samsung · Jun 17, 2024 · 130afb4 · 130afb4
1 parent 01d7bae
commit 130afb4
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 4 deletions.
diff --git a/onert-micro/eval-driver/TrainingDriver.cpp b/onert-micro/eval-driver/TrainingDriver.cpp
@@ -196,7 +196,7 @@ int entry(int argc, char **argv)
     onert_micro::OMTrainingContext train_context;
     train_context.batch_size = BATCH_SIZE;
     train_context.num_of_train_layers = num_train_layers;
-    train_context.lambda = lambda;
+    train_context.learning_rate = lambda;
     train_context.loss = loss;
     train_context.optimizer = train_optim;
     train_context.beta = beta;

diff --git a/onert-micro/onert-micro/include/OMConfig.h b/onert-micro/onert-micro/include/OMConfig.h
@@ -78,6 +78,7 @@ struct OMTrainingContext
   float epsilon = 10e-8;
   uint32_t num_step = 0;
   uint32_t num_epoch = 0;
+  uint32_t epochs = 0;
 };
 
 /*

diff --git a/onert-micro/onert-micro/src/train/tests/BostonHousingTask.test.cpp b/onert-micro/onert-micro/src/train/tests/BostonHousingTask.test.cpp
@@ -50,7 +50,7 @@ class BostonHousingTaskTest : public ::testing::Test
       onert_micro::OMTrainingContext train_context;
       train_context.batch_size = batch_size;
       train_context.num_of_train_layers = num_train_layers;
-      train_context.lambda = lambda;
+      train_context.learning_rate = lambda;
       train_context.loss = loss;
       train_context.optimizer = train_optim;
       train_context.beta = beta;

diff --git a/onert-micro/onert-micro/src/train/train_optimizers/Adam.cpp b/onert-micro/onert-micro/src/train/train_optimizers/Adam.cpp
@@ -241,7 +241,7 @@ OMStatus Adam::updateWeights(const onert_micro::OMTrainingContext &training_conf
       return UnknownError;
 
     float *f_weight_data = reinterpret_cast<float *>(weight_data);
-    float lambda = training_config.lambda;
+    float lambda = training_config.learning_rate;
     float num_step = static_cast<float>(training_config.num_step);
     float beta_in_pow_batch = std::pow(beta, num_step);
     float beta_square_in_pow_batch = std::pow(beta_squares, num_step);

diff --git a/onert-micro/onert-micro/src/train/train_optimizers/SGD.cpp b/onert-micro/onert-micro/src/train/train_optimizers/SGD.cpp
@@ -110,7 +110,7 @@ OMStatus SGD::updateWeights(const onert_micro::OMTrainingContext &training_confi
       return UnknownError;
 
     float *f_weight_data = reinterpret_cast<float *>(weight_data);
-    float lambda = training_config.lambda;
+    float lambda = training_config.learning_rate;
     const uint32_t batch_size = training_config.batch_size;
     for (uint32_t i = 0; i < flat_size; ++i)
     {