From: Inki Dae <inki.dae@samsung.com>
Date: Tue, 15 Feb 2022 01:42:00 +0000 (+0900)
Subject: make sure to clean up data set
X-Git-Tag: submit/tizen/20220307.024724~3
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=4b879d4cf7c01aebc6e41db208d0eeb7b72fe087;p=platform%2Fupstream%2Ftraining-engine-nntrainer.git

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 fdffb66..90e63bc 100644
--- 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 9151f05..61af208 100644
--- 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.");