#include <OMConfig.h>

Data Fields
uint32_t	batch_size = 1

uint32_t	num_of_train_layers = 0

OMTrainOptimizer	optimizer = SGD

OMLoss	loss = MSE

float	learning_rate = 0.001f

float	beta = 0.9f

float	beta_squares = 0.9f

float	epsilon = 10e-8

uint32_t	num_step = 0

uint32_t	num_epoch = 0

uint32_t	epochs = 0

char *	training_config_info_data = nullptr

Detailed Description

Definition at line 74 of file OMConfig.h.

Field Documentation

◆ batch_size

uint32_t onert_micro::OMTrainingContext::batch_size = 1

Definition at line 76 of file OMConfig.h.

Referenced by onert.experimental.train.dataloader.DataLoader::__init__(), entry(), onert.experimental.train.dataloader.DataLoader::split(), nnfw_session::train_get_loss(), nnfw_session::train_run(), nnfw_session::train_set_traininfo(), onert_micro::train::optimizers::Adam::updateWeights(), and onert_micro::train::optimizers::SGD::updateWeights().

◆ beta

float onert_micro::OMTrainingContext::beta = 0.9f

Definition at line 81 of file OMConfig.h.

Referenced by entry(), nnfw_session::train_set_traininfo(), and onert_micro::train::optimizers::Adam::updateWeights().

◆ beta_squares

float onert_micro::OMTrainingContext::beta_squares = 0.9f

Definition at line 82 of file OMConfig.h.

Referenced by entry(), nnfw_session::train_set_traininfo(), and onert_micro::train::optimizers::Adam::updateWeights().

◆ epochs

uint32_t onert_micro::OMTrainingContext::epochs = 0

Definition at line 86 of file OMConfig.h.

Referenced by entry().

◆ epsilon

float onert_micro::OMTrainingContext::epsilon = 10e-8

Definition at line 83 of file OMConfig.h.

Referenced by onert.experimental.train.optimizer.adam.Adam::__init__(), entry(), and onert_micro::train::optimizers::Adam::updateWeights().

◆ learning_rate

float onert_micro::OMTrainingContext::learning_rate = 0.001f

Definition at line 80 of file OMConfig.h.

Referenced by onert.experimental.train.optimizer.optimizer.Optimizer::__init__(), entry(), nnfw_session::train_set_traininfo(), onert_micro::train::optimizers::Adam::updateWeights(), and onert_micro::train::optimizers::SGD::updateWeights().

◆ loss

OMLoss onert_micro::OMTrainingContext::loss = MSE

Definition at line 79 of file OMConfig.h.

Referenced by onert.experimental.train.session.TrainSession::__init__(), onert.experimental.train.session.TrainSession::_print_training_parameters(), entry(), onert.experimental.train.session.TrainSession::eval_step(), onert.experimental.train.session.TrainSession::train(), nnfw_session::train_get_loss(), and onert.experimental.train.session.TrainSession::train_step().

◆ num_epoch

uint32_t onert_micro::OMTrainingContext::num_epoch = 0

Definition at line 85 of file OMConfig.h.

Referenced by nnfw_session::train_run().

◆ num_of_train_layers

uint32_t onert_micro::OMTrainingContext::num_of_train_layers = 0

Definition at line 77 of file OMConfig.h.

Referenced by onert_micro::import::OMExecutionPlanCreator::createBackwardExecutionPlan(), onert_micro::import::OMExecutionPlanCreator::createForwardExecutionPlan(), entry(), and nnfw_session::train_set_traininfo().

◆ num_step

uint32_t onert_micro::OMTrainingContext::num_step = 0

Definition at line 84 of file OMConfig.h.

Referenced by nnfw_session::train_run(), and onert_micro::train::optimizers::Adam::updateWeights().

◆ optimizer

OMTrainOptimizer onert_micro::OMTrainingContext::optimizer = SGD

Definition at line 78 of file OMConfig.h.

Referenced by onert.experimental.train.session.TrainSession::__init__(), onert.experimental.train.session.TrainSession::_print_training_parameters(), entry(), onert.experimental.train.session.TrainSession::eval_step(), onert.experimental.train.session.TrainSession::train(), nnfw_session::train_set_traininfo(), and onert.experimental.train.session.TrainSession::train_step().

◆ training_config_info_data

char* onert_micro::OMTrainingContext::training_config_info_data = nullptr

Definition at line 88 of file OMConfig.h.

The documentation for this struct was generated from the following file:

onert-micro/onert-micro/include/OMConfig.h

Data Fields