make sure to clean up data set

[Version] : 0.1.1-0
[Issue type] : bug fix

Make sure to clean up data pushed to two vectors - _train_data_set
and _verity_data_set at DestroyDatset function.

Change-Id: I57e1dec33f5bd9f87818e43bcd88e9f32a50924a
Signed-off-by: Inki Dae <inki.dae@samsung.com>

diff --git a/packaging/training-engine-nntrainer.spec b/packaging/training-engine-nntrainer.spec
index fdffb6654d3a371c86f68af4cd0ae61e6b6005ed..90e63bc4f9c07d4f70f8b123ceff7f77410b711c 100644 (file)
--- a/packaging/training-engine-nntrainer.spec
+++ b/packaging/training-engine-nntrainer.spec
@@ -1,6 +1,6 @@
 Name:       training-engine-nntrainer
 Summary:    Training engine NNTrainer backend
-Version:    0.1.0
+Version:    0.1.1
 Release:    0
 Group:      Multimedia/Libraries
 License:    Apache-2.0

diff --git a/src/training_engine_nntrainer.cpp b/src/training_engine_nntrainer.cpp
index 9151f0598f011454c3fe26b688a7b9435c827c02..61af208ef1681ff8c88dcbbbe493d94bf2d05422 100644 (file)
--- a/src/training_engine_nntrainer.cpp
+++ b/src/training_engine_nntrainer.cpp
@@ -420,6 +420,11 @@ namespace NntrainerImpl
 
                TE_CHECK_OBJ_NO_RETURN_VAL(dataset);
 
+               _train_data_idx = 0;
+               _verify_data_idx = 0;
+               _train_data_sets.clear();
+               _verify_data_sets.clear();
+
                int ret = ml_train_dataset_destroy(static_cast<ml_train_dataset_h>(dataset->dataset_handle));
                if (ret != ML_ERROR_NONE) {
                        LOGE("Failed to destroy a dataset handle.");