Now TFLITE Interpreter is not support loss : cross type
So in Resnet Application we made some macro to make them mse and there was some wrong part
in ResNet Application there was another macro for ENABLE_TEST GTEST's result assume that Application use cross loss
For Correct Result Fix some #if statement
TODO : even if fix this situation TEST still failed regardless of tflite export releated code
Signed-off-by: DongHak Park <donghak.park@samsung.com>
/// @todo update createResnet18 to be more generic
ModelHandle createResnet18() {
/// @todo support "LOSS : cross" for TF_Lite Exporter
/// @todo update createResnet18 to be more generic
ModelHandle createResnet18() {
/// @todo support "LOSS : cross" for TF_Lite Exporter
-#if defined(ENABLE_TEST)
+#if (defined(ENABLE_TFLITE_INTERPRETER) && !defined(ENABLE_TEST))
ModelHandle model = ml::train::createModel(ml::train::ModelType::NEURAL_NET,
{withKey("loss", "mse")});
#else
ModelHandle model = ml::train::createModel(ml::train::ModelType::NEURAL_NET,
{withKey("loss", "mse")});
#else
model->train();
#if defined(ENABLE_TEST)
model->train();
#if defined(ENABLE_TEST)
- model->exports(ml::train::ExportMethods::METHOD_TFLITE, "resnet_test.tflite");
training_loss = model->getTrainingLoss();
validation_loss = model->getValidationLoss();
training_loss = model->getTrainingLoss();
validation_loss = model->getValidationLoss();
+#elif defined(ENABLE_TFLITE_INTERPRETER)
+ model->exports(ml::train::ExportMethods::METHOD_TFLITE, "resnet_test.tflite");