#include <TrainConfigData.h>

Public Member Functions
	TrainResult ()=default

	TrainResult (TrainData train_data)

Data Fields
std::pair< onert_micro::OMMetrics, float >	best_metrics_results = {}

size_t	peak_memory_footprint = 0

Detailed Description

Definition at line 73 of file TrainConfigData.h.

Constructor & Destructor Documentation

◆ TrainResult() [1/2]

training_configure_tool::TrainResult::TrainResult ( )

default

◆ TrainResult() [2/2]

training_configure_tool::TrainResult::TrainResult ( TrainData train_data )

inlineexplicit

Definition at line 79 of file TrainConfigData.h.

  {
    peak_memory_footprint = std::numeric_limits<size_t>::max();
    switch (train_data.metrics_to_check_best_config)
    {
      case onert_micro::ACCURACY:
        best_metrics_results = {onert_micro::ACCURACY, 0.f};
        break;
      case onert_micro::CROSS_ENTROPY_METRICS:
        best_metrics_results = {onert_micro::CROSS_ENTROPY_METRICS,
                                std::numeric_limits<float>::max()};
        break;
      case onert_micro::MSE_METRICS:
        best_metrics_results = {onert_micro::MSE_METRICS, std::numeric_limits<float>::max()};
        break;
      case onert_micro::MAE_METRICS:
        best_metrics_results = {onert_micro::MAE_METRICS, std::numeric_limits<float>::max()};
        break;
      default:
        assert(false && "Unsupported type");
        break;
    }
  }

References onert_micro::ACCURACY, best_metrics_results, onert_micro::CROSS_ENTROPY_METRICS, onert_micro::MAE_METRICS, training_configure_tool::TrainData::metrics_to_check_best_config, onert_micro::MSE_METRICS, and peak_memory_footprint.

Field Documentation

◆ best_metrics_results

std::pair<onert_micro::OMMetrics, float> training_configure_tool::TrainResult::best_metrics_results = {}

Definition at line 75 of file TrainConfigData.h.

75{};

Referenced by training_configure_tool::cmpTrainResults(), training_configure_tool::findBestSparseBackpropagationTensorsRanks(), training_configure_tool::findBestTrainableOpIndexes(), training_configure_tool::runTrainProcessWithCurConfig(), and TrainResult().

◆ peak_memory_footprint

size_t training_configure_tool::TrainResult::peak_memory_footprint = 0

Definition at line 76 of file TrainConfigData.h.

Referenced by training_configure_tool::cmpTrainResults(), training_configure_tool::findBestSparseBackpropagationTensorsRanks(), training_configure_tool::findBestTrainableOpIndexes(), training_configure_tool::runTrainProcessWithCurConfig(), training_configure_tool::selectOpIndexesSetsAccordingToMemoryRestriction(), and TrainResult().

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

onert-micro/training-configure-tool/include/TrainConfigData.h

Public Member Functions